Searched defs:dim_value (Results 1 – 6 of 6) sorted by relevance
66 tensorflow::Safe_PyObjectPtr dim_value( in cast() local
57 int64_t dim_value = quotient % input_shape.dim_size(dim); in LinearSubIndexToFullIndex() local
638 Output dim_value = Slice(scope, casted_params_shape, {dim - 1}, {1}); in GetBatchIndices() local
560 auto dim_value = (*it).getSExtValue(); in matchAndRewrite() local
1183 Value input_value, Value dim_value) { in convertExpandDimsOp()
1167 Value dim_value = rewriter.create<tensor::DimOp>(loc, val, dim); in matchAndRewriteDynamicConv() local3120 auto dim_value = rewriter.create<arith::IndexCastOp>( in matchAndRewrite() local