Home
last modified time | relevance | path

Searched refs:dim_value (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Dpy_checkpoint_reader_wrapper.cc66 tensorflow::Safe_PyObjectPtr dim_value( in cast() local
69 tensorflow::Safe_PyObjectPtr dim_value( in cast()
72 PyList_SET_ITEM(value.get(), i, dim_value.release()); in cast()
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator.py461 dim_value = tensor_shape.dimension_value(self.domain_dimension)
462 if dim_value is not None:
463 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
505 dim_value = tensor_shape.dimension_value(self.range_dimension)
506 if dim_value is not None:
507 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
1010 dim_value = tensor_shape.dimension_value(self.domain_dimension)
1011 if dim_value is not None:
1012 n = dim_value
/external/tensorflow/tensorflow/core/kernels/
Dreduce_join_op.cc57 int64 dim_value = quotient % input_shape.dim_size(dim); in LinearSubIndexToFullIndex() local
59 result += strides[dim] * dim_value; in LinearSubIndexToFullIndex()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc424 auto dim_value = (*it).getSExtValue(); in matchAndRewrite() local
425 if (it == int_values.begin() && dim_value == -1) { in matchAndRewrite()
426 dim_value = 1; in matchAndRewrite()
428 new_element_shape_values.push_back(dim_value); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.h95 Value input_value, Value dim_value);
Dlegalize_common.cc1240 Value input_value, Value dim_value) { in convertExpandDimsOp() argument
1260 if (!matchPattern(dim_value, m_Constant(&dim_elem))) return llvm::None; in convertExpandDimsOp()
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py593 dim_value = casted_params_shape[dim - 1]
597 dim_indices = math_ops.range(start, dim_value, step)
600 [1] * (dim - 1) + [dim_value] + [1] * (indices_ndims - dim), axis=0)
Darray_ops.py5166 dim_value = casted_params_shape[dim - 1]
5170 dim_indices = gen_math_ops._range(start, dim_value, step)
5173 [1] * (dim - 1) + [dim_value] + [1] * (indices_ndims - dim), axis=0)