Home
last modified time | relevance | path

Searched defs:output_types (Results 1 – 25 of 108) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Diterator_ops.py101 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
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dnode_properties_test.cc64 const DataTypeVector& output_types) { in ValidateNodeProperties()
86 DataTypeVector output_types{DT_DOUBLE}; in TEST() local
116 DataTypeVector output_types{DT_DOUBLE}; in TEST() local
H A Dnode_properties.cc30 DataTypeVector output_types; in CreateFromNodeDef() local
H A Dnode_properties.h58 DataTypeVector output_types; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
H A Dsimple_tflite_op_test.cc38 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dperception_ops_utils_test.cc40 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
H A Dlstm_utils_test.cc184 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
H A Dlstm_utils.cc616 SmallVector<mlir::Type, 4> output_types; in CreateEqualSizeSplitVOp() local
808 SmallVector<Type, 5> output_types; in ConvertKerasLSTMLayer() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_send_recv.cc157 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/logging/
H A Dconversion_log_util.cc177 TFLITE_PROTO_NS::RepeatedPtrField<std::string>* output_types) { in GetInputAndOutputTypes()
253 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in PopulateConversionLog() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dwindow_dataset.cc38 Window(std::vector<std::vector<Tensor>> elements, DataTypeVector output_types, in Window()
184 DataTypeVector output_types, in NewWindow()
H A Dparallel_map_dataset_op.cc78 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()
H A Dmulti_device_iterator_ops.cc78 Env* env, const DataTypeVector& output_types, in MultiDeviceIterator()
150 const DataTypeVector& output_types() const { return output_types_; } in output_types() function in tensorflow::data::__anon7578bfd70111::MultiDeviceIterator
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/
H A Dinput_lib.py211 def output_types(self): member in DistributedIteratorV1
368 def output_types(self): member in _SingleWorkerDatasetIterator
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_rewrite_pass.cc443 llvm::SmallVector<Type, 4> output_types; in BuildExecuteOp() local
526 llvm::SmallVector<Type, 4> output_types; in AddToParallelExecuteOp() local
617 const auto output_types = cluster_func.getResultTypes(); in BuildSingletonParallelExecuteOp() local
H A Dtpu_merge_variables_with_execute.cc345 int AppendTypes(llvm::SmallVectorImpl<Type>* output_types, in AppendTypes()
369 llvm::SmallVector<Type, 8> output_types; in ReplaceParallelExecute() local
H A Dtpu_extract_outside_compilation.cc73 llvm::SmallVector<Type, 4> output_types; in BuildFunction() local
179 TypeRange output_types, Value compilation_key, in CreateRecvAtHostOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
H A Draise_to_tf.cc376 const SmallVectorImpl<Type>& output_types, in CreateAndReplaceOp()
447 SmallVector<Type, 4> output_types; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Drebatch_dataset_op.cc58 const int64_t num_replicas, const DataTypeVector& output_types, in Dataset()
321 const DataTypeVector& output_types, in Dataset()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dxla_sharding_util.cc88 llvm::SmallVector<mlir::Type, 4> output_types(num_split, output_type); in CreateSplitOp() local
514 llvm::SmallVectorImpl<mlir::Type>* output_types, in GetOutputTypesForLogicalDeviceComputation()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Ddecode_proto_ops.cc37 std::vector<tensorflow::DataType> output_types; in __anon17460a200102() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dreaders.py1172 def __init__(self, driver_name, data_source_name, query, output_types): argument
1207 def __init__(self, driver_name, data_source_name, query, output_types): argument
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dtrivial_spmd_expander.cc38 auto output_types = llvm::to_vector<8>(terminator_op.getOperandTypes()); in ExpandOp() local
/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/
H A Dexecution_trace_observer.cpp239 const std::string& output_types = "[]", in writeJsonNode()
710 std::vector<std::string> output_types; in onFunctionExit() local
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/
H A Dtfprof_options.cc44 std::set<string> output_types(kOutput, in ParseOutput() local

12345