| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | linalg_grad_test.py | 62 def _GetMatrixUnaryFunctorGradientTest(functor_, dtype_, shape_, **kwargs_): argument
103 shape_, argument
151 shape_, argument
|
| D | matrix_band_part_op_test.py | 46 def _GetMatrixBandPartTest(dtype_, batch_shape_, shape_): argument
76 def _GetMatrixBandPartGradTest(dtype_, batch_shape_, shape_): argument
|
| D | svd_op_test.py | 90 def _GetSvdOpTest(dtype_, shape_, use_static_shape_, compute_uv_, argument
227 def _GetSvdGradOpTest(dtype_, shape_, compute_uv_, full_matrices_): argument
271 def _GetSvdGradGradOpTest(dtype_, shape_, compute_uv_, full_matrices_): argument
|
| D | self_adjoint_eig_op_test.py | 123 def _GetSelfAdjointEigTest(dtype_, shape_, compute_v_): argument
185 def _GetSelfAdjointEigGradTest(dtype_, shape_, compute_v_): argument
|
| D | qr_op_test.py | 82 def _GetQrOpTest(dtype_, shape_, full_matrices_, use_static_shape_): argument
192 def _GetQrGradOpTest(dtype_, shape_, full_matrices_): argument
|
| D | eig_op_test.py | 127 def _GetEigTest(dtype_, shape_, compute_v_): argument
196 def _GetEigGradTest(dtype_, shape_, compute_v_): argument
|
| D | normalize_op_test.py | 61 def _GetNormalizeOpTest(dtype_, shape_, ord_, axis_): argument
|
| D | norm_op_test.py | 67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | variable_ops.h | 38 TensorShape shape_; variable
|
| D | immutable_constant_op.h | 43 TensorShape shape_; variable
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | variable.h | 69 TensorShape shape_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.h | 91 Shape shape_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_param_op.cc | 46 xla::Shape shape_; member in tensorflow::XlaFakeParamOp
|
| D | sendrecv_ops.cc | 60 xla::Shape shape_; member in tensorflow::__anonc9c086e80111::RecvOp
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | unary_elementwise_tester.h | 62 std::vector<int32_t> shape_; variable
|
| D | leaky_relu_tester.h | 61 std::vector<int32_t> shape_; variable
|
| D | softmax_tester.h | 63 std::vector<int32_t> shape_; variable
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | conjugate_gradient_test.py | 41 def _get_conjugate_gradient_test(dtype_, use_static_shape_, shape_): argument
|
| /external/libtextclassifier/native/utils/ |
| D | tensor-view.h | 71 const std::vector<int> shape_; variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.h | 102 BHWDC shape_; variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.h | 114 BHWDC shape_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.h | 121 const Shape shape_; variable
206 Shape shape_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.h | 372 const TensorShape shape_; variable
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.h | 205 mutable absl::optional<std::vector<int64_t>> shape_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_resource.h | 172 TensorShape shape_; variable
|