| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_3d_test.py | 231 output_sizes, argument
276 output_sizes = test_util.NHWCToNCHW(output_sizes)
291 output_sizes,
323 output_sizes=[1, 3, 3, 3, 1],
333 output_sizes=[1, 1, 2, 3, 2],
343 output_sizes=[1, 2, 1, 6, 1],
353 output_sizes=[1, 2, 2, 2, 1],
363 output_sizes=[2, 1, 1, 1, 1],
373 output_sizes=[1, 3, 2, 4, 1],
383 output_sizes=[1, 2, 1, 2, 1],
[all …]
|
| D | pooling_ops_test.py | 99 output_sizes = [[32, 35, 35, 192], [32, 17, 17, 288], [32, 8, 8, 1248],
105 for o in output_sizes:
108 for n, i, f, o, s, p in zip(names, input_sizes, filter_sizes, output_sizes,
1111 output_sizes, argument
1136 assert input_sizes[0] == output_sizes[0]
1137 assert input_sizes[3] == output_sizes[3]
1180 output_sizes,
1189 output_sizes, argument
1214 assert input_sizes[0] == output_sizes[0]
1215 assert input_sizes[3] == output_sizes[3]
[all …]
|
| D | conv_ops_test.py | 902 output_sizes, argument
913 x2 = self._CreateNumpyTensor(output_sizes)
921 t2 = constant_op.constant(x2, shape=output_sizes)
949 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
952 x2 = np.random.rand(*output_sizes).astype(np.float32)
962 t2 = constant_op.constant(x2, shape=output_sizes)
994 output_sizes=[1, 1, 2, 1],
1009 output_sizes=[0, 1, 2, 1],
1029 output_sizes=[1, 1, 2, 3],
1047 output_sizes=[1, 2, 3, 1],
[all …]
|
| D | depthwise_conv_op_test.py | 900 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
903 x2 = np.random.rand(*output_sizes).astype(dtype)
911 t2 = constant_op.constant(x2, shape=output_sizes)
958 def _CompareBackpropFilter(self, input_sizes, filter_sizes, output_sizes, argument
961 x2 = np.random.rand(*output_sizes).astype(dtype)
969 t2 = constant_op.constant(x2, shape=output_sizes)
|
| D | conv_ops_3d_test.py | 779 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
847 output_sizes=[1, 1, 5, 1, 1],
863 output_sizes=[1, 1, 5, 1, 1],
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 103 void Unflatten(const string& prefix, const std::vector<string>& output_sizes, in Unflatten() argument
105 for (int i = 0, end = output_sizes.size(); i < end; ++i) { in Unflatten()
106 if (!output_sizes[i].empty()) { in Unflatten()
112 strings::StrAppend(result, "[", var, "[:", output_sizes[i], "]] + ", in Unflatten()
113 var, "[", output_sizes[i], ":]"); in Unflatten()
116 output_sizes[i], "]] + ", var, "[", i, " + ", in Unflatten()
117 output_sizes[i], ":]"); in Unflatten()
148 const std::vector<string>& output_sizes);
154 void GetOutputSizesAndNumOutputsExpr(std::vector<string>* output_sizes,
158 const std::vector<string>& output_sizes,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | depthwise_conv_op_test.py | 457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
460 x2 = np.random.rand(*output_sizes).astype(np.float32)
466 t2 = array_ops.placeholder(np.float32, shape=output_sizes)
495 output_sizes, argument
500 x2 = np.random.rand(*output_sizes).astype(np.float32)
506 t2 = array_ops.placeholder(np.float32, shape=output_sizes)
563 output_sizes, stride, dilation, argument
566 x2 = np.random.rand(*output_sizes).astype(np.float32)
571 t2 = array_ops.placeholder(np.float32, shape=output_sizes)
624 output_sizes, argument
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | split_benchmark.py | 34 def build_graph(device, input_shape, output_sizes, axis): argument
51 outputs.extend(array_ops.split(inp, output_sizes, axis))
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 102 std::vector<int> output_sizes; in Init() local
126 output_sizes.emplace_back(static_cast<int>(buf_size)); in Init()
131 input_sizes.size(), input_sizes.data(), output_sizes.size(), in Init()
132 output_sizes.data()); in Init()
|
| D | soc_interface.h | 49 int* output_sizes);
|
| /external/libaom/libaom/test/ |
| D | cnn_test.cc | 2471 const int output_sizes[4] = { in TEST_F() local
2479 (output_sizes[0] + output_sizes[1] + output_sizes[2] + output_sizes[3])); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 2540 std::vector<int64> output_sizes(new_conv->shape().dimensions().begin(), in PropagateOnBackpropFilterConv() local
2543 output_sizes.erase(output_sizes.begin() + in PropagateOnBackpropFilterConv()
2547 TF_ASSIGN_OR_RETURN(new_conv, MakeReshapeHlo(output_sizes, new_conv)); in PropagateOnBackpropFilterConv()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2257 const TRT_TensorOrWeights& output_sizes, const int h_index, in GetSpatialDimsFromOutputSizes() argument
2261 const TRT_ShapedWeights& weights = output_sizes.weights(); in GetSpatialDimsFromOutputSizes()
|