| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 212 const std::vector<DataType>& out_types, in ShuffleOutputDataTypeAttribute() argument
215 for (int i = 0; i < out_types.size(); i++) { in ShuffleOutputDataTypeAttribute()
218 result[iter->second] = out_types[i]; in ShuffleOutputDataTypeAttribute()
395 std::vector<DataType> out_types; in MaybeRewriteIfNode() local
396 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "Tout", &out_types)); in MaybeRewriteIfNode()
397 bool has_resource_output = std::find(out_types.begin(), out_types.end(), in MaybeRewriteIfNode()
398 DT_RESOURCE) != out_types.end(); in MaybeRewriteIfNode()
501 ShuffleOutputDataTypeAttribute(out_types, retval_index_mapping); in MaybeRewriteIfNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 132 DataTypeVector out_types = n->input_types(); in AddSymGrad() local
133 for (int i = 0, end = out_types.size(); i < end; ++i) { in AddSymGrad()
134 if (out_types[i] == DT_RESOURCE) { in AddSymGrad()
136 out_types[i] = DT_FLOAT; in AddSymGrad()
139 AddNodeAttr("Tout", out_types, &ndef); in AddSymGrad()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 1028 def _get_mlir_ssa_values(self, name_prefix, out_types): argument
1031 if not out_types:
1035 if len(out_types) == 1:
1041 out_name_suffix = ':{}'.format(len(out_types))
1042 for idx, _ in enumerate(out_types):
1055 out_types = []
1057 out_types.append(str(TFRTypes.TENSOR))
1059 self.emit('scf.if {} -> ({}) {{'.format(cond, ', '.join(out_types)))
1079 return list(zip(ret_ssa_values, out_types))
1141 out_types = []
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops_test.cc | 37 std::vector<DataType> out_types; in TEST() local
40 out_types.push_back(DT_FLOAT); in TEST()
45 .Attr("OUT_TYPE", out_types) in TEST()
|
| /external/tensorflow/tensorflow/python/kernel_tests/proto/ |
| D | encode_proto_op_test_base.py | 139 out_types = [f.dtype for f in fields]
146 output_types=out_types)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 342 std::vector<DataType> out_types; in CreateTRTNode() local
371 if (out_types.size() <= conn.port_number) { in CreateTRTNode()
372 out_types.resize(conn.port_number + 1); in CreateTRTNode()
374 out_types.at(conn.port_number) = conn.connection_type; in CreateTRTNode()
498 .Attr("OutT", out_types) in CreateTRTNode()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cast_op_test.py | 207 out_types = dtypes.int8, dtypes.uint8, dtypes.int16, dtypes.float32
209 for out_type in out_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()
71 DataTypeVectorString(out_types)); in ExpectSuccess()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 266 `_ArrayToList` | `out_types={bool,complex64,double,float,int32,int64,uint32,…
|
| D | cpu_supported_ops.md | 270 `_ArrayToList` | `out_types={bool,complex64,double,float,int32,int64,uint32,…
|
| /external/boringssl/src/ssl/ |
| D | ssl_lib.cc | 2417 size_t SSL_get0_certificate_types(const SSL *ssl, const uint8_t **out_types) { in SSL_get0_certificate_types() argument
2422 *out_types = types.data(); in SSL_get0_certificate_types()
|
| /external/rust/crates/quiche/deps/boringssl/src/ssl/ |
| D | ssl_lib.cc | 2344 size_t SSL_get0_certificate_types(const SSL *ssl, const uint8_t **out_types) { in SSL_get0_certificate_types() argument
2349 *out_types = types.data(); in SSL_get0_certificate_types()
|
| /external/rust/crates/quiche/deps/boringssl/src/include/openssl/ |
| D | ssl.h | 942 const uint8_t **out_types);
|
| /external/boringssl/src/include/openssl/ |
| D | ssl.h | 942 const uint8_t **out_types);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 18286 TF_DerivedResultTypeListAttr out_types = TF_DerivedResultTypeListAttr<0>;
|