Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/core/framework/
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
Dnode_properties.cc30 DataTypeVector output_types; in CreateFromNodeDef() local
Dnode_properties.h57 DataTypeVector output_types; member
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py83 def __init__(self, iterator_resource, initializer, output_types, argument
124 def from_structure(output_types, argument
219 output_types, argument
481 def output_types(self): member in Iterator
784 def output_types(self): member in OwnedIterator
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils_test.cc38 const SmallVector<mlir::Type, NOutput>& output_types) { in createMaxUnpoolingFunc()
58 SmallVector<mlir::Type, 1> output_types{output_type}; in createMaxUnpoolingFunc() local
145 SmallVector<mlir::Type, 1> output_types{output_type}; in TEST_F() local
Dlstm_utils_test.cc183 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/
Ddecode_proto_ops.cc37 std::vector<tensorflow::DataType> output_types; in __anon955959310102() local
Dremote_fused_graph_ops_test.cc40 DataTypeVector output_types; in TEST() local
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dsql_dataset_test_base.py35 output_types, argument
/external/tensorflow/tensorflow/core/kernels/data/
Dwindow_dataset.cc31 DataTypeVector output_types, in WindowDataset()
132 DataTypeVector output_types, in NewWindowDataset()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc42 DataTypeVector output_types({DT_FLOAT}); in TEST_F() local
56 DataTypeVector output_types({DT_FLOAT}); in TEST_F() local
113 DataTypeVector output_types{DT_FLOAT}; in BuildRemoteFusedGraphExecuteOp() local
Dops_testutil.cc221 const DataTypeVector& OpsTestBase::output_types() const { in output_types() function in tensorflow::OpsTestBase
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util.cc174 TFLITE_PROTO_NS::RepeatedPtrField<std::string>* output_types) { in GetInputAndOutputTypes()
250 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in PopulateConversionLog() local
Dconversion_log_util_test.cc152 TFLITE_PROTO_NS::RepeatedPtrField<std::string> input_types, output_types; in TEST() local
/external/tensorflow/tensorflow/core/profiler/
Dtfprof_options.cc44 std::set<string> output_types(kOutput, in ParseOutput() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/sql/
Dsqlite_query_connection.cc34 const DataTypeVector& output_types) { in Open()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging.cc112 DataTypeVector output_types; in InsertLogging() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc360 const SmallVectorImpl<Type>& output_types, in CreateAndReplaceOp()
431 SmallVector<Type, 4> output_types; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dnon_serializable_dataset_op.cc44 const DataTypeVector& output_types, in Dataset()
Dassert_cardinality_dataset_op.cc42 const DataTypeVector& output_types, in Dataset()
Dassert_next_dataset_op.cc38 const DataTypeVector& output_types, in Dataset()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpost_quantize.cc104 llvm::SmallVector<Type, 4> output_types; in RemoveQuantizationAdaptorOps() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc135 DataTypeVector output_types; in BuildFusedGraphDef() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc316 int AppendTypes(llvm::SmallVectorImpl<Type>* output_types, in AppendTypes()
340 llvm::SmallVector<Type, 8> output_types; in ReplaceParallelExecute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops_test.cc95 std::vector<DataType> output_types = {DT_INT32}; in WhileWithIfBody() local

1234