Searched defs:dim_value (Results 1 – 10 of 10) sorted by relevance
660 uint64_t dim_value; in ValidateTensorDim() local744 uint64_t dim_value; in ValidateTypeTensorViewNV() local
104 uint64_t dim_value; in ValidateTensorTypeWithDimValuesNV() local
2381 uint64_t dim_value; in ValidateCooperativeMatrixLoadStoreTensorNV() local
66 tensorflow::Safe_PyObjectPtr dim_value( in cast() local
57 int64_t dim_value = quotient % input_shape.dim_size(dim); in LinearSubIndexToFullIndex() local
683 const auto dim_value = in __anonc321d2750602() 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
1167 Value dim_value = rewriter.create<tensor::DimOp>(loc, val, dim); in matchAndRewriteDynamicConv() local3120 auto dim_value = rewriter.create<arith::IndexCastOp>( in matchAndRewrite() local
1183 Value input_value, Value dim_value) { in convertExpandDimsOp()