Home
last modified time | relevance | path

Searched defs:in_types (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dforward_type_inference.cc172 auto in_types = input_types(*n); in Run() local
194 auto in_types = input_types(*n); in Run() local
Dgradients.cc117 DataTypeVector in_types; in AddSymGrad() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc41 const std::vector<DataType>& in_types, in ShuffleInputDataTypeAttribute()
55 Status InputTypesNeedsRearrange(const std::vector<DataType>& in_types, in InputTypesNeedsRearrange()
388 std::vector<DataType> in_types; in MaybeRewriteIfNode() local
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder_test.cc64 DataTypeVector in_types, out_types; in ExpectSuccess() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc431 std::vector<DataType> in_types; in TRTEngineOp() local