Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc120 const Array4D<float>& pointwise_weights, in SeparableConvArray4D() argument
124 CHECK_EQ(pointwise_weights.depth(), input.depth() * depth_multiplier); in SeparableConvArray4D()
128 Array4D<float> weights(pointwise_weights.planes(), input.depth(), in SeparableConvArray4D()
133 for (int64 out = 0; out < pointwise_weights.planes(); ++out) { in SeparableConvArray4D()
138 pointwise_weights(out, depth + kz * depth_multiplier, 0, 0); in SeparableConvArray4D()
Dreference_util.h110 const Array4D<float>& pointwise_weights,
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dlarge_function_graph.pbtxt431 name: "InceptionV2/Conv2d_1a_7x7/pointwise_weights/read"
484 input: "InceptionV2/Conv2d_1a_7x7/pointwise_weights/read"