Home
last modified time | relevance | path

Searched defs:output_sizes (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Ddepthwise_conv_op_test.py457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
495 output_sizes, argument
563 output_sizes, stride, dilation, argument
624 output_sizes, argument
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen.cc103 void Unflatten(const string& prefix, const std::vector<string>& output_sizes, in Unflatten()
418 std::vector<string> output_sizes(num_outs_); in Code() local
527 const string& function_setup, const std::vector<string>& output_sizes) { in HandleGraphMode()
772 std::vector<string>* output_sizes, string* num_outputs_expr) { in GetOutputSizesAndNumOutputsExpr()
812 const string& indentation, const std::vector<string>& output_sizes, in AddEagerFunctionTeardown()
847 const string& parameters, const std::vector<string>& output_sizes, in AddEagerFastPathAndGraphCode()
895 const string& parameters, const std::vector<string>& output_sizes, in AddEagerFallbackCode()
1005 std::vector<string> output_sizes; in AddEagerInferredAttrs() local
/external/tensorflow/tensorflow/python/ops/
Dsplit_benchmark.py34 def build_graph(device, input_shape, output_sizes, axis): argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py902 output_sizes, argument
949 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1108 output_sizes, argument
1151 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1288 output_sizes, strides, dilations, argument
1334 output_sizes, strides, dilations, argument
1532 output_sizes, argument
1694 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
Dpooling_ops_3d_test.py231 output_sizes, argument
Dpooling_ops_test.py1111 output_sizes, argument
1189 output_sizes, argument
1523 expected_input_backprop, input_sizes, output_sizes, argument
Dconv_ops_3d_test.py779 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc102 std::vector<int> output_sizes; in Init() local
/external/libaom/libaom/test/
Dcnn_test.cc2471 const int output_sizes[4] = { in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc2540 std::vector<int64> output_sizes(new_conv->shape().dimensions().begin(), in PropagateOnBackpropFilterConv() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2257 const TRT_TensorOrWeights& output_sizes, const int h_index, in GetSpatialDimsFromOutputSizes()