Searched refs:conv_input (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version_test.cc | 30 const std::string conv_input = "conv_input"; in TEST() local
33 conv->inputs.push_back(conv_input); in TEST()
37 array_map[conv_input] = std::unique_ptr<Array>(new Array); in TEST()
38 array_map[conv_input]->data_type = ArrayDataType::kFloat; in TEST()
63 const std::string conv_input = "conv_input"; in TEST() local
66 conv->inputs.push_back(conv_input); in TEST()
70 array_map[conv_input] = std::unique_ptr<Array>(new Array); in TEST()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | conv_with_shared_weights.py | 63 conv_input = input_tensor + 0.1
68 conv_input,
75 conv_input,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 188 StringPiece /*type_string*/, xla::XlaOp conv_input, xla::XlaOp filter, in MakeXlaForwardConvOp() argument
192 auto* builder = conv_input.builder(); in MakeXlaForwardConvOp()
193 TF_ASSIGN_OR_RETURN(xla::Shape input_shape, builder->GetShape(conv_input)); in MakeXlaForwardConvOp()
279 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
287 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
|
| D | conv_op_helpers.h | 61 StringPiece type_string, xla::XlaOp conv_input, xla::XlaOp filter,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1743 auto& conv_input = op_context.op_info.inputs(0); in PredictFusedConv2DBiasActivation() local
1752 conv_input.shape(), filter.shape(), op_context.op_info, in PredictFusedConv2DBiasActivation()
1763 FusedChildContext(op_context, "Conv2D", output, {conv_input, filter}), in PredictFusedConv2DBiasActivation()
|