Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/framework/
Dtensor.h44 class Tensor; variable
192 explicit Tensor(float scalar_value) in Tensor() function
194 explicit Tensor(double scalar_value) in Tensor() function
196 explicit Tensor(int32_t scalar_value) in Tensor() function
198 explicit Tensor(uint32 scalar_value) in Tensor() function
200 explicit Tensor(uint16 scalar_value) in Tensor() function
202 explicit Tensor(uint8 scalar_value) in Tensor() function
204 explicit Tensor(int16_t scalar_value) in Tensor() function
206 explicit Tensor(int8_t scalar_value) in Tensor() function
208 explicit Tensor(tstring scalar_value) in Tensor() function
[all …]
/external/tensorflow/tensorflow/go/
Dtensor.go71 type Tensor struct { struct
72 c *C.TF_Tensor
73 shape []int64
217 func (t *Tensor) finalize() { C.TF_DeleteTensor(t.c) }
220 func (t *Tensor) DataType() DataType { return DataType(C.TF_TensorType(t.c)) }
223 func (t *Tensor) Shape() []int64 { return t.shape }
226 func (t *Tensor) Reshape(newShape []int64) error {
259 func (t *Tensor) Value() interface{} {
374 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
/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
59 extension Tensor { extension
110 extension Tensor { extension
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py314 vars_: Union[List[core_types.Tensor],
653 def AddValue(self, val: core_types.Tensor) -> core_types.Tensor:
898 inputs: Optional[List[List[core_types.Tensor]]] = None,
1014 inputs: Iterable[core_types.Tensor],
1190 inputs: Optional[List[List[core_types.Tensor]]] = None,
1727 inputs: Optional[List[List[Optional[core_types.Tensor]]]] = None,
1890 inputs: Optional[List[core_types.Tensor]] = None,
1976 inputs: Optional[List[List[Optional[core_types.Tensor]]]] = None,
2038 inputs: Optional[List[List[Optional[core_types.Tensor]]]] = None,
2195 inputs: Optional[List[core_types.Tensor]] = None,
/external/ComputeLibrary/arm_compute/runtime/
DPyramid.h37 class Tensor; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dpropagator_debug_utils.h22 class Tensor; variable
Deval_const_tensor.h28 class Tensor; variable
Dstep_stats_collector.h39 class Tensor; variable
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding.h22 class Tensor; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor.cc302 Tensor::Tensor(cl_mem memory, bool memory_owner, const BHWC& shape, in Tensor() function in tflite::gpu::cl::Tensor
310 Tensor::Tensor(cl_mem memory, bool memory_owner, const BHWDC& shape, in Tensor() function in tflite::gpu::cl::Tensor
318 Tensor::Tensor(cl_mem memory, bool memory_owner, cl_mem image_buffer_memory, in Tensor() function in tflite::gpu::cl::Tensor
332 Tensor::Tensor(cl_mem memory, bool memory_owner, cl_mem image_buffer_memory, in Tensor() function in tflite::gpu::cl::Tensor
346 Tensor::Tensor(Tensor&& tensor) in Tensor() function in tflite::gpu::cl::Tensor
/external/ComputeLibrary/arm_compute/graph/detail/
DExecutionHelpers.h37 class Tensor; variable
/external/ComputeLibrary/arm_compute/graph/
DIDeviceBackend.h42 class Tensor; variable
DINode.h42 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.py96 def workload_tensors_to_ndarray(workload_tensors: List[Tuple[int, Union[Tensor, ConstTensor]]]) -> …
Dtensor.py12 class Tensor(annTensor): class
/external/tensorflow/tensorflow/compiler/jit/
Dget_compiler_ir.h26 class Tensor; variable
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py4149 def func(x: ops.Tensor):
4170 def func(*args: ops.Tensor):
4198 def func(t: ops.Tensor, **kwargs: ops.Tensor):
4218 def func(t: ops.Tensor, *args: ops.Tensor, **kwargs: ops.Tensor):
4238 def func(t: ops.Tensor, i: int = 1, **kwargs): # pylint: disable=bad-whitespace
4252 def func(x, y: ops.Tensor, z: int):
4268 def func(x, y, *args: ops.Tensor):
4283 def func(x, y, *args, **kwargs: ops.Tensor):
4303 x: ops.Tensor = 0, # pylint:disable=bad-whitespace
4305 **kwargs: ops.Tensor):
[all …]
/external/tensorflow/tensorflow/c/eager/
Dc_api.h459 class Tensor; variable
/external/ComputeLibrary/src/graph/
DTensor.cpp30 Tensor::Tensor(TensorID id, TensorDescriptor desc) in Tensor() function in arm_compute::graph::Tensor
/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc46 Status HandleElementToSlice(const Tensor& /* element */, T* src, T* dest, in HandleElementToSlice()
80 Status HandleElementToSlice<ResourceHandle>(const Tensor& /* element */, in HandleElementToSlice()
89 Status HandleElementToSlice<Eigen::half>(const Tensor& /* element */, in HandleElementToSlice()

123