Home
last modified time | relevance | path

Searched defs:broadcast_dim (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbatch_matmul.h31 inline int broadcast_dim(int lhs_dim, int rhs_dim) { in broadcast_dim() function
/external/tensorflow/tensorflow/lite/kernels/
Dbatch_matmul.cc109 int broadcast_dim = lhs_dim; in ResizeOutputTensor() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc897 for (int64_t broadcast_dim : broadcast_dims) { in ComputeArrayForElementwiseBinaryOp() local
Ddynamic_padder.cc1632 HloInstruction* broadcast_dim = dus->AddInstruction( in RewriteDynamicUpdateSlice() local
Ddynamic_dimension_inference.cc270 int64_t broadcast_dim = hlo->dimensions(dimension); in HandleBroadcast() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4665 int64_t broadcast_dim; in matchAndRewrite() local