Searched refs:depthwise_weights (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 119 const Array4D<float>& depthwise_weights, in SeparableConvArray4D() argument
123 const int64 depth_multiplier = depthwise_weights.planes(); in SeparableConvArray4D()
129 depthwise_weights.height(), depthwise_weights.width()); in SeparableConvArray4D()
130 for (int64 kx = 0; kx < depthwise_weights.width(); ++kx) { in SeparableConvArray4D()
131 for (int64 ky = 0; ky < depthwise_weights.height(); ++ky) { in SeparableConvArray4D()
137 depthwise_weights(depth, kz, ky, kx) * in SeparableConvArray4D()
|
| D | reference_util.h | 109 const Array4D<float>& input, const Array4D<float>& depthwise_weights,
|
| /external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/ |
| D | large_function_graph.pbtxt | 439 name: "InceptionV2/Conv2d_1a_7x7/depthwise_weights/read"
543 input: "InceptionV2/Conv2d_1a_7x7/depthwise_weights/read"
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | quantize_test.py | 538 depthwise_weights = variables.model_variable(
543 input1, depthwise_weights, strides, padding='SAME')
546 conv1, depthwise_weights, strides, padding='SAME')
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 1281 depthwise_weights = array_ops.tile(weights, [1, 1, num_filters_in, 1])
1282 outputs = nn.depthwise_conv2d(inputs, depthwise_weights,
2805 depthwise_weights = variables.model_variable(
2820 depthwise_weights,
|