| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_3d_test.py | 211 output_sizes, argument
256 output_sizes = test_util.NHWCToNCHW(output_sizes)
271 output_sizes,
303 output_sizes=[1, 3, 3, 3, 1],
313 output_sizes=[1, 1, 2, 3, 2],
323 output_sizes=[1, 2, 1, 6, 1],
333 output_sizes=[1, 2, 2, 2, 1],
343 output_sizes=[2, 1, 1, 1, 1],
353 output_sizes=[1, 3, 2, 4, 1],
363 output_sizes=[1, 2, 1, 2, 1],
[all …]
|
| D | pooling_ops_test.py | 92 output_sizes = [[32, 35, 35, 192], [32, 17, 17, 288], [32, 8, 8, 1248],
98 for o in output_sizes:
101 for n, i, f, o, s, p in zip(names, input_sizes, filter_sizes, output_sizes,
947 output_sizes, argument
972 assert input_sizes[0] == output_sizes[0]
973 assert input_sizes[3] == output_sizes[3]
1014 output_sizes,
1023 output_sizes, argument
1048 assert input_sizes[0] == output_sizes[0]
1049 assert input_sizes[3] == output_sizes[3]
[all …]
|
| D | conv_ops_test.py | 706 output_sizes, argument
717 x2 = self._CreateNumpyTensor(output_sizes)
724 t2 = constant_op.constant(x2, shape=output_sizes)
752 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
755 x2 = np.random.rand(*output_sizes).astype(np.float32)
765 t2 = constant_op.constant(x2, shape=output_sizes)
797 output_sizes=[1, 1, 2, 1],
812 output_sizes=[0, 1, 2, 1],
832 output_sizes=[1, 1, 2, 3],
850 output_sizes=[1, 2, 3, 1],
[all …]
|
| D | depthwise_conv_op_test.py | 523 def _CompareBackpropInputFloat(self, input_sizes, filter_sizes, output_sizes, argument
526 x2 = np.random.rand(*output_sizes).astype(np.float32)
532 t2 = constant_op.constant(x2, shape=output_sizes)
543 def _CompareBackpropInputDouble(self, input_sizes, filter_sizes, output_sizes, argument
546 x2 = np.random.rand(*output_sizes).astype(np.float64)
552 t2 = constant_op.constant(x2, shape=output_sizes)
575 def _CompareBackpropFilterFloat(self, input_sizes, filter_sizes, output_sizes, argument
578 x2 = np.random.rand(*output_sizes).astype(np.float32)
584 t2 = constant_op.constant(x2, shape=output_sizes)
596 output_sizes, stride, padding): argument
[all …]
|
| D | conv_ops_3d_test.py | 683 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
750 output_sizes=[1, 1, 5, 1, 1],
765 output_sizes=[1, 1, 5, 1, 1],
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 75 void Unflatten(const string& prefix, const std::vector<string>& output_sizes, in Unflatten() argument
77 for (int i = 0; i < output_sizes.size(); ++i) { in Unflatten()
78 if (!output_sizes[i].empty()) { in Unflatten()
81 if (i + 1 < output_sizes.size()) { in Unflatten()
84 strings::StrAppend(result, "[", var, "[:", output_sizes[i], "]] + ", in Unflatten()
85 var, "[", output_sizes[i], ":]"); in Unflatten()
88 output_sizes[i], "]] + ", var, "[", i, " + ", in Unflatten()
89 output_sizes[i], ":]"); in Unflatten()
124 void GetOutputSizesAndNumOutputsExpr(std::vector<string>* output_sizes,
128 const std::vector<string>& output_sizes,
[all …]
|
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| D | graph_compute_order_test.py | 85 output_sizes = {}
93 output_sizes[node.name] = output_size
105 self.assertIn(k, output_sizes)
106 self.assertEqual(output_sizes[k], v)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | depthwise_conv_op_test.py | 318 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
321 x2 = np.random.rand(*output_sizes).astype(np.float32)
327 t2 = array_ops.placeholder(np.float32, shape=output_sizes)
356 output_sizes, argument
361 x2 = np.random.rand(*output_sizes).astype(np.float32)
367 t2 = array_ops.placeholder(np.float32, shape=output_sizes)
|
| /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/contrib/hvx/hexagon_controller/src_soc_interface/ |
| D | soc_interface.c | 94 int* output_sizes) { in soc_interface_AllocateInOutNodeBuffers() argument
97 input_count, input_sizes, output_count, output_sizes); in soc_interface_AllocateInOutNodeBuffers()
|
| /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/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_soc_interface/include/ |
| D | soc_interface.h | 50 int* output_sizes);
|
| /external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/include/ |
| D | hexagon_controller.h | 66 int* output_sizes);
|
| /external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/ |
| D | hexagon_controller.c | 283 int* output_sizes) { in hexagon_controller_AllocateMultipleNodeDataBuffers() argument
291 hexagon_controller_AllocateOutputNodeDataBuffers(i, output_sizes[i]); in hexagon_controller_AllocateMultipleNodeDataBuffers()
|