Searched refs:in_types (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 41 const std::vector<DataType>& in_types, in ShuffleInputDataTypeAttribute() argument
44 for (int i = 0, end = in_types.size(); i < end; i++) { in ShuffleInputDataTypeAttribute()
45 result[index_mapping.at(i)] = in_types[i]; in ShuffleInputDataTypeAttribute()
55 Status InputTypesNeedsRearrange(const std::vector<DataType>& in_types, in InputTypesNeedsRearrange() argument
59 for (int i = 0, end = in_types.size(); i < end; i++) { in InputTypesNeedsRearrange()
60 DataType type = in_types[i]; in InputTypesNeedsRearrange()
73 for (int i = first_resource_index + 1, end = in_types.size(); i < end; i++) { in InputTypesNeedsRearrange()
74 if (in_types[i] != DT_RESOURCE) { in InputTypesNeedsRearrange()
84 for (int i = 0, end = in_types.size(); i < end; i++) { in InputTypesNeedsRearrange()
85 DataType type = in_types[i]; in InputTypesNeedsRearrange()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 117 DataTypeVector in_types; in AddSymGrad() local
120 in_types.push_back(nout.dtype()); in AddSymGrad()
124 in_types.push_back(nout.dtype()); in AddSymGrad()
128 AddNodeAttr("Tin", in_types, &ndef); in AddSymGrad()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cast_op_test.py | 206 in_types = dtypes.float32,
208 for in_type in in_types:
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_builder_test.cc | 63 DataTypeVector in_types, out_types; in ExpectSuccess() local
65 InOutTypesForNode(node_def, builder.op_def(), &in_types, &out_types); in ExpectSuccess()
69 DataTypeVectorString(in_types)); in ExpectSuccess()
|