| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| test_util.h | 34 DataTypeVector arg_types;
35 DataTypeVector ret_types;
|
| functionalize_control_flow_test.cc | 140 EXPECT_EQ(DataTypeVector{DT_INT32}, result.ret_types);
141 EXPECT_EQ((DataTypeVector{DT_BOOL, DT_INT32, DT_INT32}), result.arg_types);
163 EXPECT_EQ(DataTypeVector{DT_INT32}, result.ret_types);
164 EXPECT_EQ((DataTypeVector{DT_BOOL, DT_INT32, DT_INT32}), result.arg_types);
275 EXPECT_EQ(DataTypeVector{DT_INT32}, result.arg_types);
276 EXPECT_EQ(DataTypeVector{DT_BOOL}, result.ret_types);
296 EXPECT_EQ(DataTypeVector{DT_INT32}, result.arg_types);
297 EXPECT_EQ(DataTypeVector{DT_INT32}, result.ret_types);
377 EXPECT_EQ(DataTypeVector{DT_INT32}, result.arg_types);
378 EXPECT_EQ(DataTypeVector{DT_BOOL}, result.ret_types)
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_ops_definitions.h | 34 int GetOpIdFor(const string& op_type, const DataTypeVector& dt) const final;
38 using DataTypeToOp = std::tuple<DataTypeVector, SupportedOpType>;
43 const string& op_type, const DataTypeVector& dt_vec,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| window_dataset.h | 42 DataTypeVector output_types,
|
| window_dataset.cc | 23 DataTypeVector output_types,
35 const DataTypeVector& output_dtypes() const override { return output_types_; }
82 const DataTypeVector output_types_;
89 DataTypeVector output_types,
|
| sql_dataset_ops.cc | 81 const string& query, const DataTypeVector& output_types,
95 const DataTypeVector& output_dtypes() const override {
146 const DataTypeVector output_types_;
149 DataTypeVector output_types_;
|
| scan_dataset_op.cc | 77 const DataTypeVector& state_types,
78 const DataTypeVector& output_types,
99 const DataTypeVector& output_dtypes() const override {
123 DataTypeVector other_arguments_types;
261 const DataTypeVector state_types_;
262 const DataTypeVector output_types_;
267 DataTypeVector state_types_;
268 DataTypeVector output_types_;
|
| map_dataset_op.cc | 63 const DataTypeVector& output_types,
82 const DataTypeVector& output_dtypes() const override {
98 DataTypeVector other_arguments_types;
176 const DataTypeVector output_types_;
181 DataTypeVector output_types_;
|
| random_dataset_op.cc | 63 const DataTypeVector& output_dtypes() const override {
64 static DataTypeVector* dtypes = new DataTypeVector({DT_INT64});
|
| range_dataset_op.cc | 57 const DataTypeVector& output_dtypes() const override {
58 static DataTypeVector* dtypes = new DataTypeVector({DT_INT64});
|
| tensor_dataset_op.cc | 62 const DataTypeVector& output_dtypes() const override { return dtypes_; }
128 DataTypeVector dtypes_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| i_remote_fused_graph_ops_definitions.h | 38 const DataTypeVector& dt) const = 0;
|
| lookup_table_init_op.cc | 55 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(),
57 DataTypeVector expected_outputs = {};
125 DataTypeVector expected_inputs = {expected_input_0, DT_STRING};
126 DataTypeVector expected_outputs = {};
|
| queue_op.h | 54 DataTypeVector component_types_;
|
| typed_queue.h | 35 TypedQueue(const int32 capacity, const DataTypeVector& component_dtypes,
49 int32 capacity, const DataTypeVector& component_dtypes,
|
| queue_base.h | 50 QueueBase(int32 capacity, const DataTypeVector& component_dtypes,
55 const DataTypeVector& component_dtypes() const override {
146 const DataTypeVector component_dtypes_;
|
| remote_fused_graph_execute_op_test.cc | 41 DataTypeVector input_types({DT_FLOAT, DT_FLOAT});
42 DataTypeVector output_types({DT_FLOAT});
55 DataTypeVector input_types({DT_INT32, DT_INT32});
56 DataTypeVector output_types({DT_FLOAT});
112 DataTypeVector input_types{DT_FLOAT};
113 DataTypeVector output_types{DT_FLOAT};
|
| /external/tensorflow/tensorflow/core/graph/ |
| subgraph.h | 34 DataTypeVector feed_types;
38 DataTypeVector fetch_types;
|
| /external/tensorflow/tensorflow/core/kernels/data/sql/ |
| sqlite_query_connection.h | 33 const DataTypeVector& output_types) override;
49 DataTypeVector output_types_;
|
| query_connection.h | 49 const DataTypeVector& output_types) = 0;
|
| /external/tensorflow/tensorflow/core/ops/ |
| remote_fused_graph_ops_test.cc | 31 DataTypeVector input_types;
40 DataTypeVector output_types;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| graph_execution_state.h | 53 DataTypeVector feed_types, DataTypeVector fetch_types)
62 DataTypeVector feed_types;
63 DataTypeVector fetch_types;
|
| function.h | 77 DataTypeVector arg_types;
78 DataTypeVector ret_types;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| build_xla_launch_ops_pass.cc | 38 const DataTypeVector& constant_dtypes, int num_resources,
39 const DataTypeVector& arg_dtypes, const DataTypeVector& result_dtypes,
76 DataTypeVector const_dtypes(node->input_types().begin(),
78 DataTypeVector arg_dtypes(
|
| /external/tensorflow/tensorflow/core/framework/ |
| queue_interface.h | 90 virtual const DataTypeVector& component_dtypes() const = 0;
|
