Searched refs:a4dlhs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 66 Array4D<float> a4dlhs(lhs.n1(), lhs.n2(), lhs.n3(), 1); in ConvArray3DGeneralDimensionsDilated() local
67 a4dlhs.Each([&](absl::Span<const int64> indices, float* value_ptr) { in ConvArray3DGeneralDimensionsDilated()
82 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, in ConvArray3DGeneralDimensionsDilated()
|