Home
last modified time | relevance | path

Searched defs:input_sizes (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tests/
Ddepthwise_conv_op_test.py457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
493 input_sizes, argument
562 def _CompareBackpropInputWithDilation(self, input_sizes, filter_sizes, argument
622 input_sizes, argument
Dpooling_ops_test.py74 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
110 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
298 input_sizes, argument
449 input_sizes, argument
Dpooling_ops_3d_test.py46 def _VerifyValues(self, pool_func, input_sizes, window, strides, padding, argument
193 input_sizes, argument
Dconv_node_name_test.py41 def _verifyNodeNameMatch(self, layer, input_sizes, filter_sizes, strides, argument
Dconv2d_test.py45 input_sizes=None, argument
239 input_sizes=None, argument
553 input_sizes=None, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_3d_test.py48 def _VerifyOneTest(self, pool_func, input_sizes, window, strides, padding, argument
89 def _VerifyValues(self, pool_func, input_sizes, window, strides, argument
230 input_sizes, argument
Dconv_ops_test.py906 input_sizes, argument
955 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1112 input_sizes, argument
1157 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1293 def _RunAndVerifyBackpropInputDilation(self, input_sizes, filter_sizes, argument
1339 def _RunAndVerifyBackpropFilterDilation(self, input_sizes, filter_sizes, argument
1536 input_sizes, argument
1698 input_sizes, argument
Dpooling_ops_test.py190 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
284 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
329 input_sizes, argument
1130 input_sizes, argument
1208 input_sizes, argument
1543 expected_input_backprop, input_sizes, output_sizes, argument
Ddepthwise_conv_op_test.py900 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
958 def _CompareBackpropFilter(self, input_sizes, filter_sizes, output_sizes, argument
Dconv_ops_3d_test.py790 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc236 std::vector<int64> input_sizes(activation->shape().dimensions().begin(), in HandleBatchGroupCount() local
583 std::vector<int64> input_sizes(activation->shape().dimensions().begin(), in HandleConvolution() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc174 Status Conv2DBackpropComputeInputShape(const Tensor& input_sizes, in Conv2DBackpropComputeInputShape()
Dconv_grad_ops_3d.cc236 const Tensor& input_sizes = context->input(0); in Compute() local
365 const Tensor& input_sizes = context->input(0); in Compute() local
1261 const Tensor& input_sizes = context->input(0); in Compute() local
Ddepthwise_conv_grad_op.cc610 const Tensor& input_sizes = context->input(0); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_ops.cc117 xla::XlaOp input_sizes = ctx->Input(0); in Compile() local
Dconv_op_helpers.cc296 const xla::PrecisionConfig* precision_config, xla::XlaOp* input_sizes) { in MakeXlaBackpropInputConvOp()
Dstrided_slice_op.cc454 auto input_sizes = input_shape.dim_sizes(); in CompileAsDynamicUpdateSlice() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc157 Node* input_sizes = in DefaultConv2DBwdInput() local
184 Node* input_sizes = in MklConv2DBwdInput() local
/external/mesa3d/src/compiler/nir/
Dnir_opcodes.py35 def __init__(self, name, output_size, output_type, input_sizes, argument
153 def opcode(name, output_size, output_type, input_sizes, input_types, argument
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc89 NodeDef* input_sizes = graph.add_node(); in TEST() local
/external/eigen/bench/tensors/
Dtensor_benchmarks.h493 Eigen::array<TensorIndex, 2> input_sizes; in convolution() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc130 Output input_sizes = in SimpleConv2DBackpropInput() local
Dlayout_optimizer_test.cc141 Output input_sizes = in SimpleConv2DBackpropInput() local
247 auto input_sizes = GetAttrValue(*input_sizes_node); in TEST_F() local
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2478 input_sizes, argument
2980 input_sizes, argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc428 auto input_sizes = rewriter.create<ConstOp>( in matchAndRewrite() local

12