Searched refs:fill_op (Results 1 – 3 of 3) sorted by relevance
216 TFE_Op* fill_op = TFE_NewOp(tfe_context_, "Fill", status_); in TEST_F() local218 TFE_OpSetAttrType(fill_op, "T", TF_FLOAT); in TEST_F()219 TFE_OpSetAttrType(fill_op, "Tshape", TF_INT32); in TEST_F()225 CheckOutputShapes(fill_op, in TEST_F()229 TFE_DeleteOp(fill_op); in TEST_F()230 fill_op = nullptr; in TEST_F()
1362 tensorflow::NodeDef* fill_op = tensorflow_graph->add_node(); in ConvertFillOperator() local1363 fill_op->set_op("Fill"); in ConvertFillOperator()1364 fill_op->set_name(src_op.outputs[0]); in ConvertFillOperator()1366 *fill_op->add_input() = src_op.inputs[0]; in ConvertFillOperator()1367 *fill_op->add_input() = src_op.inputs[1]; in ConvertFillOperator()1368 (*fill_op->mutable_attr())["index_type"].set_type( in ConvertFillOperator()1370 (*fill_op->mutable_attr())["T"].set_type( in ConvertFillOperator()
49 "fill_op.cc",