Searched defs:expanded_shape (Results 1 – 10 of 10) sorted by relevance
52 SymDimVector expanded_shape; in ensure_has_bdim() local
64 std::vector<int64_t> expanded_shape = in Expand() local
97 llvm::ArrayRef<int64_t> expanded_shape = in IsUnitDimExpansionOnly() local
133 const auto expanded_shape = std::get<0>(expanded_and_true_global_shape); in ExpandOpAndSetLayout() local
128 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
440 Tensor expanded_shape(DT_INT64, TensorShape({1 + rank})); in Compute() local
89 std::vector<int> expanded_shape(num_expanded_dims); in DenseToSparse() local
110 Shape expanded_shape = shape; in ExpandedFilterShape() local
3286 SmallVector<Attribute, 4> expanded_shape( in ExpandedShape() local
3550 auto expanded_shape = input_hlos[i]->shape(); in HandleReduce() local