Searched refs:op_dimensions (Results 1 – 3 of 3) sorted by relevance
356 op_dimensions = [381 for op_dims in [op_dimensions, unknown_op_dimensions]:396 op_dimensions = [tensor_shape.Dimension(d) for d in op_dimension_values]399 op_dimensions, inputs, -1))412 op_dimensions = [tensor_shape.Dimension(None) for _ in range(3)]417 linear_operator_util.arg_is_blockwise(op_dimensions, x, -2)422 op_dimensions = [tensor_shape.Dimension(v) for v in op_dimension_values]429 linear_operator_util.arg_is_blockwise(op_dimensions, x, -1)
142 llvm::ArrayRef<tensorflow::int64> op_dimensions);
911 llvm::ArrayRef<int64> op_dimensions) { in ConvertDimensions() argument913 dimensions.reserve(op_dimensions.size()); in ConvertDimensions()914 for (auto value : op_dimensions) dimensions.emplace_back(APInt(64, value)); in ConvertDimensions()