Home
last modified time | relevance | path

Searched refs:GetI64ElementsAttr (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc137 static DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, in GetI64ElementsAttr() function
145 static DenseIntElementsAttr GetI64ElementsAttr(ArrayAttr attr) { in GetI64ElementsAttr() function
303 GetI64ElementsAttr(slice_starts, builder), in SliceInMinorDims()
304 GetI64ElementsAttr(slice_limits, builder), in SliceInMinorDims()
305 GetI64ElementsAttr(slice_strides, builder)); in SliceInMinorDims()
341 GetI64ElementsAttr(slice_sizes, builder)); in DynamicSliceInMinorDims()
508 auto broadcast_dims = GetI64ElementsAttr({feature_dim}, &builder); in Broadcast1DToFeatureDim()
542 auto batch_dimensions = GetI64ElementsAttr( in BatchDot()
544 auto lhs_contracting_dimensions = GetI64ElementsAttr( in BatchDot()
547 auto rhs_contracting_dimensions = GetI64ElementsAttr( in BatchDot()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc48 static mlir::DenseIntElementsAttr GetI64ElementsAttr( in GetI64ElementsAttr() function
108 GetI64ElementsAttr(window_strides, &builder_), in ConvGeneralDilatedInternal()
110 GetI64ElementsAttr(lhs_dilation, &builder_), in ConvGeneralDilatedInternal()
111 GetI64ElementsAttr(rhs_dilation, &builder_), in ConvGeneralDilatedInternal()
127 GetI64ElementsAttr(fft_length, &builder_)); in FftInternal()
165 GetI64ElementsAttr(dimensions_to_reduce, &builder_)); in ReduceInternal()
192 GetI64ElementsAttr(sizes, &builder_), in ReduceWindowInternal()
193 GetI64ElementsAttr(strides, &builder_), in ReduceWindowInternal()
194 GetI64ElementsAttr(base_dilations, &builder_), in ReduceWindowInternal()
195 GetI64ElementsAttr(win_dilations, &builder_), in ReduceWindowInternal()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Doptimize_mhlo.cc44 static DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, in GetI64ElementsAttr() function
131 auto start = GetI64ElementsAttr({i}, &rewriter); in matchAndRewrite()
132 auto limit = GetI64ElementsAttr({i + 1}, &rewriter); in matchAndRewrite()
133 auto stride = GetI64ElementsAttr({1}, &rewriter); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc111 DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, in GetI64ElementsAttr() function
129 return GetI64ElementsAttr(slice_limits, builder); in BuildSliceLimits()
236 GetI64ElementsAttr(slice_start, &rewriter), in matchAndRewrite()
237 GetI64ElementsAttr(slice_end, &rewriter), in matchAndRewrite()
238 GetI64ElementsAttr(slice_stride, &rewriter)); in matchAndRewrite()
392 GetI64ElementsAttr(iota_dimension_int, &rewriter), in matchAndRewrite()
393 GetI64ElementsAttr(iota_dimension_int + 1, &rewriter), in matchAndRewrite()
394 GetI64ElementsAttr(1, &rewriter)); in matchAndRewrite()
1436 GetI64ElementsAttr(temp_start_indices, &rewriter); in matchAndRewrite()
1440 GetI64ElementsAttr(SmallVector<int64_t, 4>(input_rank, 1), &rewriter); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc42 static DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, in GetI64ElementsAttr() function
109 return GetI64ElementsAttr(dims_to_reduce, builder); in GetBiasAddGradReductionIndices()
289 loc, GetI64ElementsAttr(packed_shape, &rewriter)); in matchAndRewrite()
990 loc, GetI64ElementsAttr(permutation_vals, &rewriter)); in matchAndRewrite()
Dlower_tf.td36 NativeCodeCall<"GetI64ElementsAttr({" # value # "}, &$_builder)">;