Home
last modified time | relevance | path

Searched defs:Tensor (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/core/framework/
Dtensor.h44 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/
Dsave_restore.py39 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,
Dapi.py341 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/tensorflow/tensorflow/go/
Dtensor.go71 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/
Dsparse_conditional_accumulator.h73 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()
Dreshape_util.h26 class Tensor; variable
Dtensor_list_util.h26 class Tensor; variable
Dvariant_ops_util.h26 class Tensor; variable
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensor.h328 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/
DTensor.swift19 public struct Tensor: Equatable, Hashable { struct
60 extension Tensor { extension
111 extension Tensor { extension
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/
Dquantize_model_test.py76 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]:
Dquantize_model_test_base.py218 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/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v1.py182 embeddings: ops.Tensor,
183 weight: ops.Tensor,
216 def _pad_or_truncate_with_sequence_length(self, embeddings: ops.Tensor,
Dtpu.py331 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/
Dpropagator_debug_utils.h22 class Tensor; variable
Deval_const_tensor.h28 class Tensor; variable
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding.h22 class Tensor; variable
/external/ComputeLibrary/arm_compute/graph/detail/
DExecutionHelpers.h37 class Tensor; variable
/external/ComputeLibrary/arm_compute/graph/
DIDeviceBackend.h42 class Tensor; variable
DINode.h43 class Tensor; variable
/external/tensorflow/tensorflow/lite/kernels/shim/
Dtensor_view.h76 explicit Tensor(TensorView *t) in Tensor() function
82 explicit Tensor(const TensorView *t) in Tensor() function
/external/ComputeLibrary/src/runtime/
DTensor.cpp28 Tensor::Tensor(IRuntimeContext *) in Tensor() function in arm_compute::Tensor
/external/armnn/python/pyarmnn/src/pyarmnn/_tensor/
Dworkload_tensors.py97 def workload_tensors_to_ndarray(workload_tensors: List[Tuple[int, Union[Tensor, ConstTensor]]]) -> …
Dtensor.py12 class Tensor(annTensor): class
/external/ComputeLibrary/src/common/utils/
DObject.h41 Tensor = 3, enumerator

1234