Home
last modified time | relevance | path

Searched refs:CreateNodeDef (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc135 NodeDef CreateNodeDef(const string& op_type, const DataTypeVector& inputs) { in CreateNodeDef() function in tensorflow::__anon74460fe50111::OpKernelTest
162 CreateNodeDef(op_type, inputs), TF_GRAPH_DEF_VERSION, &status)); in ExpectSuccess()
208 NodeDef ndef = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F()
219 NodeDef ndef = CreateNodeDef("Test3", {DT_INT8, DT_INT8}); in TEST_F()
228 NodeDef ndef = CreateNodeDef("Test3", {DT_FLOAT, DT_FLOAT}); in TEST_F()
236 NodeDef ndef = CreateNodeDef("Test3", {DT_STRING, DT_STRING}); in TEST_F()
244 NodeDef ndef = CreateNodeDef("Test4", {DT_FLOAT}); in TEST_F()
254 NodeDef ndef = CreateNodeDef("Test5", {DT_STRING, DT_STRING}); in TEST_F()
269 ExpectFailure(CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}).DebugString(), in TEST_F()
271 ExpectFailure(CreateNodeDef("Test3", {DT_INT8, DT_INT8}).DebugString(), in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.h40 static Status CreateNodeDef(const string& name, const TensorValue& tensor,
Dconstant_folding.cc645 CreateNodeDef(const_name, TensorValue(&value), out[j])); in MaterializeBroadcastGradientArgs()
758 CreateNodeDef(const_name, TensorValue(&value), reduction_indices)); in MaterializeReductionIndices()
1066 Status ConstantFolding::CreateNodeDef(const string& name, in CreateNodeDef() function in tensorflow::grappler::ConstantFolding
1211 Status s = CreateNodeDef(node_name, output_tensors[i], &outputs->at(i), in EvaluateOneFoldable()
2303 !CreateNodeDef(axis_node->name(), TensorValue(&axis_t), axis_node) in SimplifyPack()
2413 if (!CreateNodeDef(false_node->name(), TensorValue(&false_t), false_node) in SimplifySwitch()
2421 if (!CreateNodeDef(true_node->name(), TensorValue(&true_t), true_node) in SimplifySwitch()
2584 if (!CreateNodeDef(OptimizedNodeName(*node, "_shape_const"), shape_value, in SimplifyReduction()
3321 CreateNodeDef(out_const_name, TensorValue(&value), out_node)); in AddQuantizedMatMulMinMaxOutConstNodes()
Dremapper.cc712 TF_CHECK_OK(ConstantFolding::CreateNodeDef( in AddBatchNormNodes()
Darithmetic_optimizer.cc2431 status = ConstantFolding::CreateNodeDef(new_const_node->name(), value, in TrySimplify()