| /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 | 58 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()
85 std::vector<tflite::TensorType> output_types = { in TEST() local
118 std::vector<tflite::TensorType> output_types = {tflite::TensorType_INT32, in TEST() local
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 101 def __init__(self, iterator_resource, initializer, output_types, argument
149 def from_structure(output_types, argument
244 output_types, argument
511 def output_types(self): member in Iterator
824 def output_types(self): member in OwnedIterator
|
| D | structured_function.py | 300 def output_types(self): member in StructuredFunctionWrapper
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 40 const SmallVector<mlir::Type, NOutput>& output_types) { in createMaxUnpoolingFunc()
59 SmallVector<mlir::Type, 1> output_types{output_type}; in createMaxUnpoolingFunc() local
147 SmallVector<mlir::Type, 1> output_types{output_type}; in TEST_F() local
|
| D | lstm_utils_test.cc | 184 auto output_types = fused_lstm_func_.getFunctionType().getResults(); in TEST_F() local
255 auto output_types = fused_ln_lstm_func_.getFunctionType().getResults(); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | decode_proto_ops.cc | 37 std::vector<tensorflow::DataType> output_types; in __anon7ac7ec0f0102() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | dtensor_send_recv.cc | 157 llvm::SmallVector<mlir::Type, 4> output_types{output_type}; in LowerDTensorRecvToXlaOp() local
235 llvm::SmallVector<mlir::Type, 4> output_types{dtensor_recv.getType()}; in LowerDTensorRecvFromCPUToTFOp() local
|
| D | cluster_function_conversion.cc | 130 llvm::SmallVector<mlir::Type, 8> output_types{ in ReplaceClusterWithPartitionCallOp() 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()
|
| D | parallel_map_dataset_op.cc | 78 int64_t num_parallel_calls, const DataTypeVector& output_types, in Dataset()
88 int64_t num_parallel_calls, const DataTypeVector& output_types, in Dataset()
786 const DataTypeVector& output_types, in MakeDataServiceUncompressDataset()
|
| /external/tensorflow/tensorflow/lite/toco/logging/ |
| D | conversion_log_util.cc | 177 TFLITE_PROTO_NS::RepeatedPtrField<std::string>* output_types) { in GetInputAndOutputTypes()
253 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in PopulateConversionLog() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | trivial_spmd_expander.cc | 38 auto output_types = llvm::to_vector<8>(terminator_op.getOperandTypes()); in ExpandOp() local
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | input_lib.py | 211 def output_types(self): member in DistributedIteratorV1
368 def output_types(self): member in _SingleWorkerDatasetIterator
|
| /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/compiler/mlir/tfrt/transforms/ |
| D | fuse_tpu_compile_and_execute_ops.cc | 92 llvm::SmallVector<mlir::Type, 4> output_types; in runOnOperation() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 112 DataTypeVector output_types; in InsertLogging() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | group_by_dialect.cc | 119 std::vector<Type> output_types; in wrapOpsInFunction() local
|
| 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/tfr/passes/ |
| D | raise_to_tf.cc | 376 const SmallVectorImpl<Type>& output_types, in CreateAndReplaceOp()
447 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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 223 const DataTypeVector& OpsTestBase::output_types() const { in output_types() function in tensorflow::OpsTestBase
|