Searched refs:ExtractConstIntFromValue (Results 1 – 9 of 9) sorted by relevance
47 ExtractConstIntFromValue(expand_dims_op.dim())); in ExpandOp()81 ExtractConstIntFromValue(expand_dims_op.dim())); in ComputeLayoutForward()113 ExtractConstIntFromValue(expand_dims_op.dim())); in ComputeLayoutBackward()
49 ExtractConstIntFromValue(argmax_op.dimension())); in ComputeResultLayout()68 ExtractConstIntFromValue(argmax_op.dimension())); in ExpandOp()145 ExtractConstIntFromValue(argmax_op.dimension())); in ComputeLayoutBackward()
37 ExtractConstIntFromValue(concat_dim_operand)); in VerifyConcatLayout()54 ExtractConstIntFromValue(concat_dim_operand)); in ReduceForConcatOutputLayout()
37 TF_ASSIGN_OR_RETURN(int64_t axis, ExtractConstIntFromValue(gather_op.axis())); in ExpandOp()173 TF_ASSIGN_OR_RETURN(int64_t axis, ExtractConstIntFromValue(gather_op.axis())); in ComputeLayoutForward()252 ExtractConstIntFromValue(gather_op.axis())); in ComputeLayoutBackward()
46 ExtractConstIntFromValue(cumsum.axis())); in GetAxisDimension()
71 ExtractConstIntFromValue(split_dim_value)); in GetAdjustedSplitDim()
1008 const auto depth = ExtractConstIntFromValue(one_hot_op.depth()); in ExpandOp()
53 StatusOr<int64_t> ExtractConstIntFromValue(mlir::Value value);
122 StatusOr<int64_t> ExtractConstIntFromValue(mlir::Value value) { in ExtractConstIntFromValue() function