| /external/tensorflow/tensorflow/compiler/tests/ |
| D | depthwise_conv_op_test.py | 457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
495 output_sizes, argument
567 output_sizes, stride, dilation, argument
628 output_sizes, argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 103 void Unflatten(const string& prefix, const std::vector<string>& output_sizes, in Unflatten()
420 std::vector<string> output_sizes(num_outs_); in Code() local
529 const string& function_setup, const std::vector<string>& output_sizes) { in HandleGraphMode()
778 std::vector<string>* output_sizes, string* num_outputs_expr) { in GetOutputSizesAndNumOutputsExpr()
818 const string& indentation, const std::vector<string>& output_sizes, in AddEagerFunctionTeardown()
853 const string& parameters, const std::vector<string>& output_sizes, in AddEagerFastPathAndGraphCode()
903 const string& parameters, const std::vector<string>& output_sizes, in AddEagerFallbackCode()
1014 std::vector<string> output_sizes; in AddEagerInferredAttrs() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | split_benchmark.py | 30 def build_graph(device, input_shape, output_sizes, axis): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | conv_ops_test.py | 914 output_sizes, argument
961 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1137 output_sizes, argument
1180 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1317 output_sizes, strides, dilations, argument
1363 output_sizes, strides, dilations, argument
1561 output_sizes, argument
1723 output_sizes, argument
|
| D | depthwise_conv_op_base.py | 955 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1015 def _CompareBackpropFilter(self, input_sizes, filter_sizes, output_sizes, argument
|
| D | pooling_ops_3d_test.py | 251 output_sizes, argument
|
| D | pooling_ops_test.py | 1153 output_sizes, argument
1231 output_sizes, argument
1565 expected_input_backprop, input_sizes, output_sizes, argument
|
| D | conv_ops_3d_test.py | 786 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 719 llvm::SmallVector<int, 2> output_sizes; in matchAndRewrite() local
893 llvm::ArrayRef<int32_t> output_sizes, in CreateResizeBilinearOp()
913 llvm::SmallVector<int, 2> &output_sizes, in MatchResizeOp()
|
| /external/libaom/test/ |
| D | cnn_test.cc | 2483 const int output_sizes[4] = { in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 3429 std::vector<int64_t> output_sizes(new_conv->shape().dimensions().begin(), in PropagateOnBackpropFilterConv() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1774 const TRT_TensorOrWeights& output_sizes, const int h_index, in GetSpatialDimsFromOutputSizes()
|