Searched defs:dim_value (Results 1 – 13 of 13) sorted by relevance
66 tensorflow::Safe_PyObjectPtr dim_value( in cast() local
104 uint64_t dim_value; in ValidateTensorTypeWithDimValuesNV() local
660 uint64_t dim_value; in ValidateTensorDim() local744 uint64_t dim_value; in ValidateTypeTensorViewNV() local
2381 uint64_t dim_value; in ValidateCooperativeMatrixLoadStoreTensorNV() local
745 uint64_t dim_value; in ValidateTensorDim() local829 uint64_t dim_value; in ValidateTypeTensorViewNV() local
2364 uint64_t dim_value; in ValidateCooperativeMatrixLoadStoreTensorNV() local
57 int64_t dim_value = quotient % input_shape.dim_size(dim); in LinearSubIndexToFullIndex() local
683 const auto dim_value = in __anon74f36cf50602() 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