| /external/tensorflow/tensorflow/core/framework/ |
| D | node_properties_test.cc | 64 const DataTypeVector& output_types) { in ValidateNodeProperties()
86 DataTypeVector output_types{DT_DOUBLE}; in TEST() local
116 DataTypeVector output_types{DT_DOUBLE}; in TEST() local
|
| D | node_properties.cc | 30 DataTypeVector output_types; in CreateFromNodeDef() local
|
| D | node_properties.h | 57 DataTypeVector output_types; member
|
| /external/tensorflow/tensorflow/lite/kernels/shim/test_op/ |
| D | simple_tflite_op_test.cc | 38 const std::vector<tflite::TensorType>& output_types) { in SimpleOpModel()
84 std::vector<tflite::TensorType> output_types = { in TEST() local
115 std::vector<tflite::TensorType> output_types = {tflite::TensorType_INT32, in TEST() local
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 100 def __init__(self, iterator_resource, initializer, output_types, argument
141 def from_structure(output_types, argument
236 output_types, argument
499 def output_types(self): member in Iterator
824 def output_types(self): member in OwnedIterator
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 38 const SmallVector<mlir::Type, NOutput>& output_types) { in createMaxUnpoolingFunc()
57 SmallVector<mlir::Type, 1> output_types{output_type}; in createMaxUnpoolingFunc() local
144 SmallVector<mlir::Type, 1> output_types{output_type}; in TEST_F() local
|
| D | lstm_utils_test.cc | 183 auto output_types = fused_lstm_func_.getType().getResults(); in TEST_F() local
252 auto output_types = fused_ln_lstm_func_.getType().getResults(); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | decode_proto_ops.cc | 37 std::vector<tensorflow::DataType> output_types; in __anone00368560102() local
|
| /external/tensorflow/tensorflow/lite/toco/logging/ |
| D | conversion_log_util.cc | 174 TFLITE_PROTO_NS::RepeatedPtrField<std::string>* output_types) { in GetInputAndOutputTypes()
250 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in PopulateConversionLog() local
|
| D | conversion_log_util_test.cc | 152 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | window_dataset.cc | 38 Window(std::vector<std::vector<Tensor>> elements, DataTypeVector output_types, in Window()
184 DataTypeVector output_types, in NewWindow()
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | tfprof_options.cc | 44 std::set<string> output_types(kOutput, in ParseOutput() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/sql/ |
| D | sqlite_query_connection.cc | 34 const DataTypeVector& output_types) { in Open()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 112 DataTypeVector output_types; in InsertLogging() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 373 const SmallVectorImpl<Type>& output_types, in CreateAndReplaceOp()
444 SmallVector<Type, 4> output_types; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | non_serializable_dataset_op.cc | 44 const DataTypeVector& output_types, in Dataset()
|
| D | assert_cardinality_dataset_op.cc | 42 const DataTypeVector& output_types, in Dataset()
|
| D | assert_next_dataset_op.cc | 38 const DataTypeVector& output_types, in Dataset()
|
| D | rebatch_dataset_op.cc | 58 const int64_t num_replicas, const DataTypeVector& output_types, in Dataset()
321 const DataTypeVector& output_types, in Dataset()
|
| D | sql_dataset_op.cc | 81 const DataTypeVector& output_types, in Dataset()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 221 const DataTypeVector& OpsTestBase::output_types() const { in output_types() function in tensorflow::OpsTestBase
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_merge_variables_with_execute.cc | 345 int AppendTypes(llvm::SmallVectorImpl<Type>* output_types, in AppendTypes()
369 llvm::SmallVector<Type, 8> output_types; in ReplaceParallelExecute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 92 llvm::SmallVector<mlir::Type, 4> output_types(num_split, output_type); in CreateSplitOp() local
503 llvm::SmallVectorImpl<mlir::Type>* output_types) { in GetOutputTypesForLogicalDeviceComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | post_quantize.cc | 137 llvm::SmallVector<Type, 4> output_types; in RemoveQuantizationAdaptorOps() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_functional_ops_test.cc | 95 std::vector<DataType> output_types = {DT_INT32}; in WhileWithIfBody() local
|