Searched refs:start_attr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 599 FloatOrIntAtrr start_attr, in BuildConstRangeTensor() argument
602 ValueType start = start_attr.getValue(); in BuildConstRangeTensor()
659 auto start_attr = start_tensor.getValues<IntegerAttr>()[0]; in fold() local
664 uint64_t start = start_attr.getUInt(); in fold()
674 num_elements = GetLengthOfRange(start_attr.getInt(), limit_attr.getInt(), in fold()
677 return BuildConstRangeTensor(elem_type, num_elements, start_attr, in fold()
680 auto start_attr = start_tensor.getValues<FloatAttr>()[0]; in fold() local
683 const int num_elements = GetLengthOfRange(start_attr.getValueAsDouble(), in fold()
686 return BuildConstRangeTensor(elem_type, num_elements, start_attr, in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 3063 FloatOrIntAtrr start_attr, in BuildConstRangeTensor() argument
3066 ValueType start = start_attr.getValue(); in BuildConstRangeTensor()
3095 auto start_attr = start_tensor.getValues<IntegerAttr>()[0]; in fold() local
3099 start_attr.getInt(), limit_attr.getInt(), delta_attr.getInt()); in fold()
3100 return BuildConstRangeTensor(elem_type, num_elements, start_attr, in fold()
3103 auto start_attr = start_tensor.getValues<FloatAttr>()[0]; in fold() local
3106 const int num_elements = GetLengthOfRange(start_attr.getValueAsDouble(), in fold()
3109 return BuildConstRangeTensor(elem_type, num_elements, start_attr, in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 587 auto start_attr = rewriter.create<ConstOp>( in SliceNegativePadding() local
602 start_attr, size_attr); in SliceNegativePadding()
|