Searched defs:input0_dims (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
H A Dreference_ops.h109 // Dims<N> refer to shapes in reverse order. In this case, input0_dims will be
123 inline void NdArrayDescsForElementwiseBroadcast(const Dims<N>& input0_dims, argument
132 desc0_out->extents[i] = input0_dims.sizes[i];
133 desc0_out->strides[i] = input0_dims.strides[i];
142 const int extent0 = ArraySize(input0_dims, i);
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Doptimized_ops.h172 // Dims<N> refer to shapes in reverse order. In this case, input0_dims will be
186 inline void NdArrayDescsForElementwiseBroadcast(const Dims<N>& input0_dims, argument
195 desc0_out->extents[i] = input0_dims.sizes[i];
196 desc0_out->strides[i] = input0_dims.strides[i];
205 const int extent0 = ArraySize(input0_dims, i);

Completed in 156 milliseconds