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.cc141 NodeDef CreateNodeDef(const string& op_type, const DataTypeVector& inputs, in CreateNodeDef() function in tensorflow::__anonc25ee1c30111::OpKernelTest
170 CreateNodeDef(op_type, inputs), TF_GRAPH_DEF_VERSION, &status)); in ExpectSuccess()
216 NodeDef ndef = CreateNodeDef("Test1", {DT_FLOAT, DT_INT32}); in TEST_F()
230 CreateNodeDef("OpWithoutKernel", {DT_STRING, DT_STRING}, remote_device); in TEST_F()
245 CreateNodeDef("OpWithoutKernel", {DT_STRING, DT_STRING}, local_device); in TEST_F()
259 NodeDef ndef = CreateNodeDef("Test3", {DT_INT8, DT_INT8}); in TEST_F()
268 NodeDef ndef = CreateNodeDef("Test3", {DT_FLOAT, DT_FLOAT}); in TEST_F()
276 NodeDef ndef = CreateNodeDef("Test3", {DT_STRING, DT_STRING}); in TEST_F()
284 NodeDef ndef = CreateNodeDef("Test4", {DT_FLOAT}); in TEST_F()
294 NodeDef ndef = CreateNodeDef("Test5", {DT_STRING, DT_STRING}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.h43 static Status CreateNodeDef(const string& name, const TensorValue& tensor,
Dconstant_folding.cc699 CreateNodeDef(const_name, TensorValue(&value), out[j])); in MaterializeBroadcastGradientArgs()
813 CreateNodeDef(const_name, TensorValue(&value), reduction_indices)); in MaterializeReductionIndices()
1226 Status ConstantFolding::CreateNodeDef(const string& name, in CreateNodeDef() function in tensorflow::grappler::ConstantFolding
1373 Status s = CreateNodeDef(node_name, output_tensors[i], &outputs->at(i), in EvaluateOneFoldable()
1835 if (!CreateNodeDef(const_name, TensorValue(&shape_t), &tmp).ok()) { in ReplaceOperationWithBroadcastTo()
2441 !CreateNodeDef(axis_node_name, TensorValue(&axis_t), &new_node).ok()) { in SimplifyPack()
2668 if (!CreateNodeDef(tmp_false_node.name(), TensorValue(&false_t), in SimplifySwitch()
2676 if (!CreateNodeDef(tmp_true_node.name(), TensorValue(&true_t), in SimplifySwitch()
2879 if (!CreateNodeDef(OptimizedNodeName(*node, "_shape_const"), shape_value, in SimplifyReduction()
3879 CreateNodeDef(out_const_name, TensorValue(&value), out_node)); in AddQuantizedMatMulMinMaxOutConstNodes()
Dremapper.cc1554 TF_RETURN_IF_ERROR(ConstantFolding::CreateNodeDef( in IsCpuCompatibleDataType()
Darithmetic_optimizer.cc2538 status = ConstantFolding::CreateNodeDef(new_const_node->name(), value, in TrySimplify()