Home
last modified time | relevance | path

Searched defs:dtype_ (Results 1 – 25 of 83) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.h40 DataType dtype_; variable
55 DataType dtype_; variable
Dscoped_allocator_ops.cc77 DataType dtype_; member in tensorflow::ScopedAllocatorOp
163 DataType dtype_; member in tensorflow::ScopedAllocatorConcatOp
229 DataType dtype_; member in tensorflow::ScopedAllocatorSplitOp
Dimmutable_constant_op.h42 DataType dtype_; variable
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinalg_grad_test.py58 def _GetMatrixUnaryFunctorGradientTest(functor_, dtype_, shape_, **kwargs_): argument
98 dtype_, argument
146 dtype_, argument
Dsvd_op_test.py120 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
Dself_adjoint_eig_op_test.py119 def _GetSelfAdjointEigTest(dtype_, shape_, compute_v_): argument
181 def _GetSelfAdjointEigGradTest(dtype_, shape_, compute_v_): argument
Dqr_op_test.py78 def _GetQrOpTest(dtype_, shape_, full_matrices_, use_static_shape_): argument
188 def _GetQrGradOpTest(dtype_, shape_, full_matrices_): argument
Deig_op_test.py135 def _GetEigTest(dtype_, shape_, compute_v_): argument
204 def _GetEigGradTest(dtype_, shape_, compute_v_): argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateless_random_ops.cc176 DataType dtype_; member in tensorflow::__anon9e8111950411::StatelessRandomUniformOp
227 DataType dtype_; member in tensorflow::__anon9e8111950411::StatelessRandomUniformIntOp
268 DataType dtype_; member in tensorflow::__anon9e8111950411::StatelessRandomUniformFullIntOp
319 DataType dtype_; member in tensorflow::__anon9e8111950411::StatelessRandomNormalOp
364 DataType dtype_; member in tensorflow::__anon9e8111950411::StatelessTruncatedNormalOp
427 DataType dtype_; member in tensorflow::__anon9e8111950411::StatelessParameterizedTruncatedNormalOp
Dtensor_array_ops.cc182 DataType dtype_; member in tensorflow::__anon9bf2a2b60111::TensorArrayOp
238 DataType dtype_; member in tensorflow::__anon9bf2a2b60111::TensorArrayWriteOp
282 DataType dtype_; member in tensorflow::__anon9bf2a2b60111::TensorArrayReadOp
349 DataType dtype_; member in tensorflow::__anon9bf2a2b60111::TensorArrayGatherOp
436 DataType dtype_; member in tensorflow::__anon9bf2a2b60111::TensorArrayScatterOp
476 DataType dtype_; member in tensorflow::__anon9bf2a2b60111::TensorArrayConcatOp
546 DataType dtype_; member in tensorflow::__anon9bf2a2b60111::TensorArraySplitOp
Dstack_ops.cc117 DataType dtype_; member in tensorflow::__anon81b44f530111::StackOp
167 DataType dtype_; member in tensorflow::__anon81b44f530111::StackPushOp
222 DataType dtype_; member in tensorflow::__anon81b44f530111::StackPopOp
Dtensor_list_ops.cc203 DataType dtype_; member in tensorflow::__anonbe250eb30111::TensorListReserveOp
277 DataType dtype_; member in tensorflow::__anonbe250eb30111::EmptyTensorListOp
378 DataType dtype_; member in tensorflow::__anonbe250eb30111::TensorListGetItemOp
431 DataType dtype_; member in tensorflow::__anonbe250eb30111::TensorListGatherOp
572 DataType dtype_; member in tensorflow::__anonbe250eb30111::TensorListSplitOp
695 DataType dtype_; member in tensorflow::__anonbe250eb30111::TensorListPopBackOp
Dtraining_ops.cc111 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyProximalGradientDescent
386 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyProximalAdagrad
463 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyAdagradDA
558 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyAdam
631 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyAdaMax
733 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyRMSProp
878 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyFtrl
900 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyFtrlV2
969 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplyAdadelta
1034 DataType dtype_; member in tensorflow::__anon9de8e2990111::ResourceApplySignBase
Dstateful_random_ops.cc247 DataType dtype_; member in tensorflow::__anon0d2456050111::StatefulUniformOp
286 DataType dtype_; member in tensorflow::__anon0d2456050111::StatefulStandardNormalOp
332 DataType dtype_; member in tensorflow::__anon0d2456050111::StatefulTruncatedNormalOp
367 DataType dtype_; member in tensorflow::__anon0d2456050111::StatefulUniformIntOp
400 DataType dtype_; member in tensorflow::__anon0d2456050111::StatefulUniformFullIntOp
Dretval_op.cc63 DataType dtype_; member in tensorflow::__anonfb13f8f80111::RetvalOp
Dstateless_random_ops_v2.cc265 DataType dtype_; member in tensorflow::__anon180fb01c0211::StatelessRandomUniformOp
329 DataType dtype_; member in tensorflow::__anon180fb01c0211::StatelessRandomUniformIntOp
378 DataType dtype_; member in tensorflow::__anon180fb01c0211::StatelessRandomUniformFullIntOp
449 DataType dtype_; member in tensorflow::__anon180fb01c0211::StatelessRandomNormalOp
506 DataType dtype_; member in tensorflow::__anon180fb01c0211::StatelessTruncatedNormalOp
Dempty_op.cc58 DataType dtype_; member in tensorflow::__anon55ea45590111::EmptyOp
Darg_op.cc110 DataType dtype_; member in tensorflow::XlaArgOp
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dmatrix_band_part_op_test.py42 def _GetMatrixBandPartTest(dtype_, batch_shape_, shape_): argument
72 def _GetMatrixBandPartGradTest(dtype_, batch_shape_, shape_): argument
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dtensor_spec.h46 DataType dtype_; variable
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dvariable.h68 DataType dtype_; variable
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.h40 DataType dtype_; variable
/external/federated-compute/fcp/aggregation/core/
Dtensor_spec.h41 const DataType dtype_; variable
Dtensor.h116 DataType dtype_; variable
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dconjugate_gradient_test.py37 def _get_conjugate_gradient_test(dtype_, use_static_shape_, shape_): argument

1234