| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | linalg_grad_test.py | 58 def _GetMatrixUnaryFunctorGradientTest(functor_, dtype_, shape_, **kwargs_): argument
99 shape_, argument
147 shape_, argument
|
| H A D | svd_op_test.py | 120 def _GetSvdOpTest(dtype_, shape_, use_static_shape_, compute_uv_, argument
257 def _GetSvdGradOpTest(dtype_, shape_, compute_uv_, full_matrices_): argument
301 def _GetSvdGradGradOpTest(dtype_, shape_, compute_uv_, full_matrices_): argument
|
| H A D | self_adjoint_eig_op_test.py | 119 def _GetSelfAdjointEigTest(dtype_, shape_, compute_v_): argument
181 def _GetSelfAdjointEigGradTest(dtype_, shape_, compute_v_): argument
|
| H A D | qr_op_test.py | 78 def _GetQrOpTest(dtype_, shape_, full_matrices_, use_static_shape_): argument
188 def _GetQrGradOpTest(dtype_, shape_, full_matrices_): argument
|
| H A D | eig_op_test.py | 135 def _GetEigTest(dtype_, shape_, compute_v_): argument
204 def _GetEigGradTest(dtype_, shape_, compute_v_): argument
|
| H A D | normalize_op_test.py | 57 def _GetNormalizeOpTest(dtype_, shape_, ord_, axis_): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | matrix_band_part_op_test.py | 42 def _GetMatrixBandPartTest(dtype_, batch_shape_, shape_): argument
72 def _GetMatrixBandPartGradTest(dtype_, batch_shape_, shape_): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/ |
| H A D | tensor_view.h | 141 const absl::Span<int> shape_; variable
221 absl::Span<int> shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | indexed_array_analysis.h | 121 const Shape shape_; variable
207 Shape shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| H A D | converter.cc | 179 BHWC shape_; member in tflite::gpu::gl::__anonf5dd32d10111::FromTensorConverter
274 BHWC shape_; member in tflite::gpu::gl::__anonf5dd32d10111::ToTensorConverter
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | variable_ops.h | 38 TensorShape shape_; variable
|
| H A D | immutable_constant_op.h | 43 TensorShape shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| H A D | variable.h | 69 TensorShape shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | loop_emitter.h | 90 Shape shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | fake_param_op.cc | 46 xla::Shape shape_; member in tensorflow::XlaFakeParamOp
|
| H A D | sendrecv_ops.cc | 60 xla::Shape shape_; member in tensorflow::__anonc39364770111::RecvOp
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/core/ |
| H A D | tensor_spec.h | 42 const TensorShape shape_; variable
|
| H A D | tensor.h | 118 TensorShape shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | unary_elementwise_tester.h | 62 std::vector<int32_t> shape_; variable
|
| H A D | softmax_tester.h | 63 std::vector<int32_t> shape_; variable
|
| H A D | leaky_relu_tester.h | 61 std::vector<int32_t> shape_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/backend/ |
| H A D | backend_data.h | 54 Shape shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | parallel_loop_emitter.h | 89 Shape shape_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| H A D | conjugate_gradient_test.py | 37 def _get_conjugate_gradient_test(dtype_, use_static_shape_, shape_): argument
|
| /aosp_15_r20/external/libtextclassifier/native/utils/ |
| H A D | tensor-view.h | 71 const std::vector<int> shape_; variable
|