Searched refs:dim_value (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/util/ |
| D | py_checkpoint_reader_wrapper.cc | 66 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/ |
| D | linear_operator.py | 465 dim_value = tensor_shape.dimension_value(self.domain_dimension)
466 if dim_value is not None:
467 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
509 dim_value = tensor_shape.dimension_value(self.range_dimension)
510 if dim_value is not None:
511 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
1014 dim_value = tensor_shape.dimension_value(self.domain_dimension)
1015 if dim_value is not None:
1016 n = dim_value
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduce_join_op.cc | 57 int64_t dim_value = quotient % input_shape.dim_size(dim); in LinearSubIndexToFullIndex() local
59 result += strides[dim] * dim_value; in LinearSubIndexToFullIndex()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 638 Output dim_value = Slice(scope, casted_params_shape, {dim - 1}, {1}); in GetBatchIndices() local
643 Output dim_indices = Range(scope, start, Squeeze(scope, dim_value), step); in GetBatchIndices()
648 {Output(Tile(scope, one, Const(scope, {dim - 1}))), dim_value, in GetBatchIndices()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 560 auto dim_value = (*it).getSExtValue(); in matchAndRewrite() local
561 if (it == int_values.begin() && dim_value == -1) { in matchAndRewrite()
570 dim_value = 1; in matchAndRewrite()
572 new_element_shape_values.push_back(dim_value); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 96 Value input_value, Value dim_value);
|
| D | legalize_common.cc | 1183 Value input_value, Value dim_value) { in convertExpandDimsOp() argument
1203 if (!matchPattern(dim_value, m_Constant(&dim_elem))) return llvm::None; in convertExpandDimsOp()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 602 dim_value = casted_params_shape[dim - 1]
606 dim_indices = math_ops.range(start, dim_value, step)
609 array_ops.tile([1], [dim - 1]), [dim_value],
|
| D | array_ops.py | 5443 dim_value = casted_params_shape[dim - 1]
5447 dim_indices = gen_math_ops._range(start, dim_value, step)
5450 [1] * (dim - 1) + [dim_value] + [1] * (indices_ndims - dim), axis=0)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1167 Value dim_value = rewriter.create<tensor::DimOp>(loc, val, dim); in matchAndRewriteDynamicConv() local
1169 dim_value); in matchAndRewriteDynamicConv()
3120 auto dim_value = rewriter.create<arith::IndexCastOp>( in matchAndRewrite() local
3124 dim_value, end_value); in matchAndRewrite()
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 341 shapeDims.push_back(CHECKED_NON_NEGATIVE(CHECKED_INT32(onnxShape.dim(i).dim_value()))); in ToTensorInfo()
|