Home
last modified time | relevance | path

Searched refs:depthwise_weights (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc104 const Array4D<float>& depthwise_weights, in SeparableConvArray4D() argument
108 const int64 depth_multiplier = depthwise_weights.planes(); in SeparableConvArray4D()
114 depthwise_weights.height(), depthwise_weights.width()); in SeparableConvArray4D()
115 for (int64 kx = 0; kx < depthwise_weights.width(); ++kx) { in SeparableConvArray4D()
116 for (int64 ky = 0; ky < depthwise_weights.height(); ++ky) { in SeparableConvArray4D()
122 depthwise_weights(depth, kz, ky, kx) * in SeparableConvArray4D()
Dreference_util.h109 const Array4D<float>& input, const Array4D<float>& depthwise_weights,
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dlarge_function_graph.pbtxt448 name: "InceptionV2/Conv2d_1a_7x7/depthwise_weights/read"
552 input: "InceptionV2/Conv2d_1a_7x7/depthwise_weights/read"