Searched refs:InOutTypesForNode (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_compatibility_test.cc | 53 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types, in ExpectSuccess()
63 TF_ASSERT_OK(InOutTypesForNode(*node_def(), *new_op_def, &new_in_types, in ExpectSuccess()
109 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types, in ExpectInvalid()
133 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types, in ExpectTypeMismatch()
144 TF_ASSERT_OK(InOutTypesForNode(*node_def(), *new_op_def, &new_in_types, in ExpectTypeMismatch()
|
| D | memory_types.cc | 91 InOutTypesForNode(ndef, *op_def, &inp_dtypes, &out_dtypes)); in MemoryTypesForNode()
|
| D | node_def_util.cc | 448 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InOutTypesForNode() function
544 TF_RETURN_IF_ERROR(InOutTypesForNode(node_def, op_def, &inputs, &outputs)); in ValidateNodeDef()
|
| D | node_def_util.h | 273 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def,
|
| D | node_def_builder_test.cc | 65 InOutTypesForNode(node_def, builder.op_def(), &in_types, &out_types); in ExpectSuccess()
|
| D | op_kernel.cc | 1307 s.Update(InOutTypesForNode(node_def, *op_def, &inputs, &outputs)); in CreateOpKernel()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.cc | 159 InOutTypesForNode(props_->node_def, *(props_->op_def), &inputs, &outputs); in UpdateProperties()
413 status->Update(InOutTypesForNode(node_def, *op_def, &inputs, &outputs)); in AddNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 101 InOutTypesForNode(ndef, *op_def, &input_dtypes_, &output_dtypes_)); in Init()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils.cc | 591 TF_RETURN_IF_ERROR(InOutTypesForNode(node_def, *op_def, inputs, outputs)); in GetInOutTypes()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 209 TF_RETURN_IF_ERROR(InOutTypesForNode(*node, op_reg_data->op_def, &input_types, in HandleInvariantNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1078 TF_RETURN_IF_ERROR(InOutTypesForNode(*node, node_ctx.op_data->op_def, in AddNode()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 226 InOutTypesForNode(*test_def, *op_def, &input_types, &output_types)); in BuildGraph()
|