tensorflow/compiler/jit/xla_kernel_creator_test.cc
// Create a FunctionDef that takes one resource and one regular param
Create a FunctionDef that takes one resource and one regular param
Create a FunctionDef that takes one resource and one regular param
FunctionDef XTimesY() {
  return FunctionDefHelper::Define(
      "XTimesY",
      {"x: float", "y: resource"},
      {"z: float"},
      {},
      {
          {{"y0"}, "ReadVariableOp", {"y"}, {{"dtype", DT_FLOAT}}},
          {{"z"}, "Mul", {"x", "y0"}, {{"T", DT_FLOAT}}},
      });
}
FunctionDef XTimesY ( ) { return FunctionDefHelper :: Define ( " " , { " " , " " } , { " " } , { } , { { { " " } , " " , { " " } , { { " " , DT_FLOAT } } } , { { " " } , " " , { " " , " " } , { { " " , DT_FLOAT } } } , } ) ; }
Create a FunctionDef that takes one resource and one regular param
Create a FunctionDef that takes one resource and one regular param
// Name // Args // Return values // Attr def // Nodes