| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tinputs", &input_types_));
79 input_count == input_types_.size())
82 << ", type count = " << input_types_.size();
128 DataTypeVector input_types_; member in class:tensorflow::RemoteFusedGraphExecuteOp
|
| ops_testutil.h | 106 if (kernel_ != nullptr) input_types_ = kernel_->input_types();
140 CHECK_GT(input_types_.size(), inputs_.size())
210 CHECK_GT(input_types_.size(), inputs_.size())
212 bool is_ref = IsRefType(input_types_[inputs_.size()]);
216 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]), dtype);
219 CHECK_EQ(input_types_[inputs_.size()], dtype);
233 DataTypeVector input_types_; member in class:tensorflow::OpsTestBase
|
| remote_fused_graph_rewriter_transform_test.cc | 108 if (!input_types_.empty()) {
111 {input_types_}}));
165 input_types_ = "float";
204 string input_types_; member in class:tensorflow::graph_transforms::__anon39893::FuseRemoteGraphMultipleAddOpsRewriterTest
|
| resize_area_op_test.cc | 41 bool is_ref = IsRefType(input_types_[inputs_.size()]);
47 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]),
51 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v());
|
| resize_bicubic_op_test.cc | 41 bool is_ref = IsRefType(input_types_[inputs_.size()]);
47 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]),
51 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v());
|
| resize_bilinear_op_test.cc | 47 bool is_ref = IsRefType(input_types_[inputs_.size()]);
53 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]),
57 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v());
|
| set_kernels.cc | 343 input_types_(input_types) {}
355 const InputTypes input_types_; member in class:tensorflow::SetOperationOp
684 switch (input_types_) {
|
| fake_quant_ops_test.cc | 36 CHECK_GT(input_types_.size(), inputs_.size())
42 bool is_ref = IsRefType(input_types_[inputs_.size()]);
44 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]), DT_FLOAT);
47 CHECK_EQ(input_types_[inputs_.size()], DT_FLOAT);
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| tf_driver.cc | 72 input_types_.resize(input_layer.size());
77 CHECK(DataTypeFromString(input_layer_type[i], &input_types_[i]));
114 auto tensor = CreateTensor(input_types_[id], input_shapes_[id]);
115 switch (input_types_[id]) {
125 fprintf(stderr, "Unsupported type %d in SetInput\n", input_types_[id]);
135 switch (input_types_[id]) {
145 fprintf(stderr, "Unsupported type %d in ResetTensor\n", input_types_[id]);
154 CreateTensor(input_types_[id], input_shapes_[id]);
166 fprintf(stderr, "Unsupported type %d in ResetTensor\n", input_types_[id]);
|
| tf_driver.h | 64 std::vector<tensorflow::DataType> input_types_; member in class:tflite::testing::TfDriver
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.h | 40 input_types_.reserve(node->num_inputs());
42 input_types_.push_back(node->input_type(i));
54 DataType input_type(int64 idx) const { return input_types_[idx]; }
72 std::vector<DataType> input_types_; member in class:tensorflow::ExtendedInferenceContext
|
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.h | 127 int num_inputs() const { return input_types_.size(); }
128 DataType input_type(int i) const { return input_types_[i]; }
129 const DataTypeVector& input_types() const { return input_types_; }
172 const DataTypeVector input_types_; member in class:tensorflow::OpKernel
281 int num_inputs() const { return input_types_.size(); }
282 DataType input_type(int i) const { return input_types_[i]; }
283 const DataTypeSlice& input_types() const { return input_types_; }
351 DataTypeSlice input_types_; member in class:tensorflow::OpKernelConstruction
[all...] |
| op_kernel.cc | 90 input_types_(context->input_types().begin(),
196 input_types_(input_types),
213 return MatchSignatureHelper(expected_inputs, expected_outputs, input_types_,
[all...] |
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| transform.py | 154 input_types_ = op._input_types[:]
162 [], input_types_, None, op_def_)
|
