Home
last modified time | relevance | path

Searched defs:shape_ (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dmatrix_band_part_op_test.py46 def _GetMatrixBandPartTest(dtype_, batch_shape_, shape_): argument
76 def _GetMatrixBandPartGradTest(dtype_, batch_shape_, shape_): argument
Dlinalg_grad_test.py62 def _GetMatrixUnaryFunctorGradientTest(functor_, dtype_, shape_, **kwargs_): argument
103 shape_, argument
Dsvd_op_test.py92 def _GetSvdOpTest(dtype_, shape_, use_static_shape_, compute_uv_, argument
226 def _GetSvdGradOpTest(dtype_, shape_, compute_uv_, full_matrices_): argument
274 def _GetSvdGradGradOpTest(dtype_, shape_, compute_uv_, full_matrices_): argument
Dqr_op_test.py79 def _GetQrOpTest(dtype_, shape_, full_matrices_, use_static_shape_): argument
171 def _GetQrGradOpTest(dtype_, shape_, full_matrices_): argument
Dself_adjoint_eig_op_test.py122 def _GetSelfAdjointEigTest(dtype_, shape_, compute_v_): argument
184 def _GetSelfAdjointEigGradTest(dtype_, shape_, compute_v_): argument
Dnormalize_op_test.py61 def _GetNormalizeOpTest(dtype_, shape_, ord_, axis_): argument
Dnorm_op_test.py67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
Deig_op_test.py126 def _GetEigTest(dtype_, shape_, compute_v_): argument
/external/tensorflow/tensorflow/core/kernels/
Dvariable_ops.h38 TensorShape shape_; variable
Dimmutable_constant_op.h43 TensorShape shape_; variable
Dconditional_accumulator_base.h126 const PartialTensorShape shape_; variable
Dconditional_accumulator_base_op.h90 PartialTensorShape shape_; variable
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.h82 Shape shape_; variable
Dir_array.h312 Shape shape_; variable
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_param_op.cc44 xla::Shape shape_; member in tensorflow::XlaFakeParamOp
Dsendrecv_ops.cc60 xla::Shape shape_; member in tensorflow::__anon1ab2458c0111::RecvOp
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dconjugate_gradient_test.py41 def _get_conjugate_gradient_test(dtype_, use_static_shape_, shape_): argument
/external/libtextclassifier/native/utils/
Dtensor-view.h71 const std::vector<int> shape_; variable
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle_data.h50 const TensorShape shape_; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.h121 const Shape shape_; variable
206 Shape shape_; variable
/external/webrtc/webrtc/modules/desktop_capture/
Ddesktop_frame.h91 rtc::scoped_ptr<DesktopRegion> shape_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor.h142 BHWDC shape_; variable
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.h170 TensorShape shape_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter.cc174 BHWC shape_; member in tflite::gpu::gl::__anondba7f7c60111::FromTensorConverter
269 BHWC shape_; member in tflite::gpu::gl::__anondba7f7c60111::ToTensorConverter
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dxfeed_manager_test.cc57 Shape shape_; member in xla::__anonce6b4c6e0111::TestInfeedBuffer

12