res/TensorFlowLiteRecipes/MatrixSetDiag_000/test.recipe

   1 operand {
   2   name: "ifm1"
   3   type: FLOAT32
   4   shape { dim: 2 dim: 3 dim: 4 }
   5 }
   6 operand {
   7   name: "ifm2"
   8   type: FLOAT32
   9   shape { dim: 2 dim: 3 }
  10 }
  11 operand {
  12   name: "ofm"
  13   type: FLOAT32
  14   shape { dim: 2 dim: 3 dim: 4 }
  15 }
  16 operation {
  17   type: "MatrixSetDiag"
  18   matrix_set_diag_options {
  19   }
  20   input: "ifm1"
  21   input: "ifm2"
  22   output: "ofm"
  23 }
  24 input: "ifm1"
  25 input: "ifm2"
  26 output: "ofm"