| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 44 class Tensor; variable
207 explicit Tensor(float scalar_value) in Tensor() function
209 explicit Tensor(double scalar_value) in Tensor() function
211 explicit Tensor(int32_t scalar_value) in Tensor() function
213 explicit Tensor(uint32 scalar_value) in Tensor() function
215 explicit Tensor(uint16 scalar_value) in Tensor() function
217 explicit Tensor(uint8 scalar_value) in Tensor() function
219 explicit Tensor(int16_t scalar_value) in Tensor() function
221 explicit Tensor(int8_t scalar_value) in Tensor() function
223 explicit Tensor(tstring scalar_value) in Tensor() function
[all …]
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | save_restore.py | 39 tensors: List[ops.Tensor],
85 file_prefix: Union[str, ops.Tensor],
88 tensors: List[Union[ops.Tensor, tf_variables.Variable]],
156 name_tensor_dict: Dict[str, Union[ops.Tensor, tf_variables.Variable]],
227 ops.Tensor],
228 name_tensor_dict: Dict[str, Union[ops.Tensor,
|
| D | api.py | 341 def fetch_layout(tensor: ops.Tensor) -> layout_lib.Layout:
357 def check_layout(tensor: ops.Tensor, layout: layout_lib.Layout) -> None:
373 def relayout(tensor: ops.Tensor, layout: layout_lib.Layout) -> ops.Tensor:
|
| /external/federated-compute/fcp/tensorflow/ |
| D | example_selector_fuser.py | 31 def example_selector_fuser(example_selector: tf.Tensor,
32 resumption_token_type_url: tf.Tensor,
33 resumption_token_content: tf.Tensor,
|
| D | delete_file.py | 28 def delete_file(filename: tf.Tensor) -> tf.Operation:
40 def delete_dir(dirname: tf.Tensor, recursively: bool = False) -> tf.Operation:
|
| D | crc32.py | 30 def crc32(tensor: tf.Tensor, name: Optional[str] = None) -> tf.Operation:
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 71 type Tensor struct { struct
72 c *C.TF_Tensor
73 shape []int64
212 func (t *Tensor) finalize() { C.TF_DeleteTensor(t.c) }
215 func (t *Tensor) DataType() DataType { return DataType(C.TF_TensorType(t.c)) }
218 func (t *Tensor) Shape() []int64 { return t.shape }
221 func (t *Tensor) Reshape(newShape []int64) error {
254 func (t *Tensor) Value() interface{} {
369 func (t *Tensor) WriteContentsTo(w io.Writer) (int64, error) {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_conditional_accumulator.h | 73 std::tuple<const Tensor*, const Tensor*, const Tensor*>* tensor, in ValidateShape()
148 std::tuple<const Tensor*, const Tensor*, const Tensor*>* grad) override { in AllocateAndAssignToAccumGradFunction()
176 std::tuple<const Tensor*, const Tensor*, const Tensor*>* grad) override { in AddToAccumGradFunction()
333 std::tuple<const Tensor*, const Tensor*, const Tensor*>** tensor) override in GetAndValidateTensorInputForApplyGrad()
379 void CleanUpGradTensor(std::tuple<const Tensor*, const Tensor*, in CleanUpGradTensor()
|
| D | tensor_list_util.h | 26 class Tensor; variable
|
| D | reshape_util.h | 26 class Tensor; variable
|
| D | variant_ops_util.h | 26 class Tensor; variable
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | Tensor.h | 328 EIGEN_STRONG_INLINE Tensor() in Tensor() function
334 EIGEN_STRONG_INLINE Tensor(const Self& other) in Tensor() function
341 … EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index firstDimension, IndexTypes... otherDimensions) in Tensor() function
348 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE explicit Tensor(Index dim1) in Tensor() function
353 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2) in Tensor() function
358 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3) in Tensor() function
363 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4) in Tensor() function
368 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4, Index… in Tensor() function
376 … EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE explicit Tensor(const array<Index, NumIndices>& dimensions) in Tensor() function
384 EIGEN_STRONG_INLINE Tensor(const TensorBase<OtherDerived, ReadOnlyAccessors>& other) in Tensor() function
[all …]
|
| /external/tensorflow/tensorflow/lite/swift/Sources/ |
| D | Tensor.swift | 19 public struct Tensor: Equatable, Hashable { struct
60 extension Tensor { extension
111 extension Tensor { extension
|
| /external/federated-compute/fcp/aggregation/core/ |
| D | tensor.h | 48 Tensor(Tensor&& other) in Tensor() function
68 Tensor() : dtype_(DT_INVALID), shape_{}, data_(nullptr) {} in Tensor() function
112 Tensor(DataType dtype, TensorShape shape, std::unique_ptr<TensorData> data) in Tensor() function
|
| /external/federated-compute/fcp/artifact_building/ |
| D | tensor_utils.py | 77 filename: Union[tf.Tensor, str],
79 tensors: list[tf.Tensor],
115 filename: Union[tf.Tensor, str],
|
| D | checkpoint_utils.py | 201 tensor_specs: Iterable[tf.TensorSpec], filepath_tensor: tf.Tensor
392 tff_structure: Union[tff.structure.Struct, tf.Tensor], name: str = 'v'
458 tf.Tensor,
479 tff_structure: Union[tff.structure.Struct, tf.Tensor],
|
| D | proto_helpers.py | 27 t: tf.Tensor, shape_hint: Optional[tf.TensorShape] = None
65 t: tf.Tensor, name: str, tff_type: tff.types.TensorType
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| D | quantize_model_test.py | 76 def matmul(self, matmul_input: core.Tensor) -> Mapping[str, core.Tensor]:
93 def conv(self, conv_input: core.Tensor) -> Mapping[str, core.Tensor]:
128 def __call__(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
245 def conv(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
354 def __call__(self, x: core.Tensor) -> Mapping[str, core.Tensor]:
449 def conv(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
556 def conv(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
950 def model_fn(self, x: core.Tensor) -> Mapping[str, core.Tensor]:
|
| D | quantize_model_test_base.py | 218 inputs: Mapping[str, core.Tensor],
304 input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
319 def conv(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
371 def matmul(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
|
| /external/federated-compute/fcp/demo/ |
| D | federated_context.py | 165 tf.Tensor,
233 input_state: Union[tff.structure.Struct, tf.Tensor,
252 self, structure: Union[tff.structure.Struct, tf.Tensor,
272 tf.Tensor]) -> bytes:
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v1.py | 182 embeddings: ops.Tensor,
183 weight: ops.Tensor,
216 def _pad_or_truncate_with_sequence_length(self, embeddings: ops.Tensor,
|
| D | tpu.py | 331 vars_: Union[List[core_types.Tensor],
671 def AddValue(self, val: core_types.Tensor) -> core_types.Tensor:
917 inputs: Optional[List[List[core_types.Tensor]]] = None,
1033 inputs: Iterable[core_types.Tensor],
1209 inputs: Optional[List[List[core_types.Tensor]]] = None,
1782 inputs: Optional[List[List[Optional[core_types.Tensor]]]] = None,
1946 inputs: Optional[List[core_types.Tensor]] = None,
2033 inputs: Optional[List[List[Optional[core_types.Tensor]]]] = None,
2096 inputs: Optional[List[List[Optional[core_types.Tensor]]]] = None,
2253 inputs: Optional[List[core_types.Tensor]] = None,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | propagator_debug_utils.h | 22 class Tensor; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.h | 22 class Tensor; variable
|
| /external/ComputeLibrary/arm_compute/graph/detail/ |
| D | ExecutionHelpers.h | 37 class Tensor; variable
|