Searched refs:permuted_shape (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/utils/ |
| D | GraphUtils.cpp | 50 arm_compute::TensorShape permuted_shape = shape; in compute_permutation_parameters() local
58 arm_compute::permute(permuted_shape, perm_shape); in compute_permutation_parameters()
61 return std::make_pair(permuted_shape, perm); in compute_permutation_parameters()
274 TensorShape permuted_shape = tensor.info()->tensor_shape(); in access_tensor() local
278 …std::tie(permuted_shape, perm) = compute_permutation_parameters(tensor.info()->tensor_shape(), ten… in access_tensor()
281 …UTE_EXIT_ON_MSG_VAR(image_loader->width() != permuted_shape.x() || image_loader->height() != permu… in access_tensor()
284 … static_cast<uint64_t>(permuted_shape.x()), static_cast<uint64_t>(permuted_shape.y())); in access_tensor()
351 TensorShape permuted_shape = tensor.info()->tensor_shape(); in access_tensor() local
355 … std::tie(permuted_shape, perm) = compute_permutation_parameters(tensor.info()->tensor_shape(), in access_tensor()
359 …ARM_COMPUTE_EXIT_ON_MSG_VAR(jpeg.width() != permuted_shape.x() || jpeg.height() != permuted_shape.… in access_tensor()
[all …]
|
| D | Utils.h | 405 TensorShape permuted_shape = tensor.info()->tensor_shape(); in fill_tensor() local
412 arm_compute::permute(permuted_shape, perm_vec); in fill_tensor()
419 … ARM_COMPUTE_ERROR_ON_MSG(permuted_shape[i] != _shape[i], "Tensor dimensions mismatch"); in fill_tensor()
444 permuted_shape.set(dim, _shape[num_dims - dim - 1]); in fill_tensor()
463 window.use_tensor_dimensions(permuted_shape); in fill_tensor()
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClSoftmax.cpp | 111 …const TensorShape permuted_shape = misc::shape_calculator::compute_permutation_output_sh… in validate() local
112 TensorInfo input_permuted(src.clone()->set_tensor_shape(permuted_shape)); in validate()
114 TensorInfo output_permuted(dst.clone()->set_tensor_shape(permuted_shape)); in validate()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuSoftmax.cpp | 144 …const TensorShape permuted_shape = misc::shape_calculator::compute_permutation_output_sh… in validate() local
145 TensorInfo input_permuted(src->clone()->set_tensor_shape(permuted_shape)); in validate()
147 TensorInfo output_permuted(dst->clone()->set_tensor_shape(permuted_shape)); in validate()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 997 Shape permuted_shape = ShapeUtil::PermuteDimensions(permutation, shape()); in Transpose() local
1012 CHECK(LayoutUtil::IsDenseArray(permuted_shape)); in Transpose()
1013 Layout* layout = permuted_shape.mutable_layout(); in Transpose()
1018 Literal new_literal(permuted_shape); in Transpose()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 940 auto permuted_shape = Shape(scope, permuted); in ProdGrad() local
987 auto y = Reshape(scope, Mul(scope, left, right), permuted_shape); in ProdGrad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 298 permuted_shape = array_ops.shape(permuted)
306 math_ops.conj(left) * math_ops.conj(right), permuted_shape)
|