| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 44 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 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 scalar_value) in Tensor() function
206 explicit Tensor(int8 scalar_value) in Tensor() function
208 explicit Tensor(tstring scalar_value) in Tensor() function
[all …]
|
| D | tensor.cc | 643 Tensor::Tensor() : Tensor(DT_FLOAT) {} in Tensor() function in tensorflow::Tensor
645 Tensor::Tensor(DataType type) : shape_(type), buf_(nullptr) {} in Tensor() function in tensorflow::Tensor
647 Tensor::Tensor(DataType type, const TensorShape& shape, TensorBuffer* buf) in Tensor() function in tensorflow::Tensor
653 Tensor::Tensor(DataType type, const TensorShape& shape, in Tensor() function in tensorflow::Tensor
764 Tensor::Tensor(Allocator* a, DataType type, const TensorShape& shape) in Tensor() function in tensorflow::Tensor
777 Tensor::Tensor(Allocator* a, DataType type, const TensorShape& shape, in Tensor() function in tensorflow::Tensor
806 Tensor::Tensor(DataType type, const TensorShape& shape) in Tensor() function in tensorflow::Tensor
|
| D | shape_inference_testutil.h | 32 class Tensor; variable
|
| /external/tensorflow/tensorflow/python/tf_program/tests/ |
| D | mlir_gen_test.py | 53 def test_fn(x: core.Tensor) -> core.Tensor:
119 def test_fn(x: core.Tensor, y: core.Tensor, z: core.Tensor):
148 def test_fn(x: core.Tensor) -> int:
181 def test_fn(x: core.Tensor) -> core.Tensor:
206 def test_fn(x: core.Tensor) -> core.Tensor:
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 71 type Tensor struct { struct
72 c *C.TF_Tensor
73 shape []int64
209 func (t *Tensor) finalize() { C.TF_DeleteTensor(t.c) }
212 func (t *Tensor) DataType() DataType { return DataType(C.TF_TensorType(t.c)) }
215 func (t *Tensor) Shape() []int64 { return t.shape }
218 func (t *Tensor) Reshape(newShape []int64) error {
251 func (t *Tensor) Value() interface{} {
366 func (t *Tensor) WriteContentsTo(w io.Writer) (int64, error) {
|
| D | tensor_handle.go | 46 func NewTensorHandle(t *Tensor) (*TensorHandle, error) {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_conditional_accumulator.h | 85 std::tuple<const Tensor*, const Tensor*, const Tensor*>* tensor, in ValidateShape()
160 std::tuple<const Tensor*, const Tensor*, const Tensor*>* grad) override { in AllocateAndAssignToAccumGradFunction()
194 std::tuple<const Tensor*, const Tensor*, const Tensor*>* grad) override { in AddToAccumGradFunction()
358 std::tuple<const Tensor*, const Tensor*, const Tensor*>** tensor) override in GetAndValidateTensorInputForApplyGrad()
404 void CleanUpGradTensor(std::tuple<const Tensor*, const Tensor*, in CleanUpGradTensor()
|
| D | reshape_util.h | 26 class Tensor; variable
|
| D | gather_nd_op.h | 32 class Tensor; variable
|
| /external/tensorflow/tensorflow/lite/swift/Sources/ |
| D | Tensor.swift | 19 public struct Tensor: Equatable, Hashable { struct
59 extension Tensor { extension
110 extension Tensor { extension
|
| /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/python/tpu/ |
| D | tpu.py | 648 def AddValue(self, val: core_types.Tensor) -> core_types.Tensor:
883 inputs: Optional[List[List[core_types.Tensor]]] = None,
999 inputs: Iterable[core_types.Tensor],
1176 inputs: Optional[List[List[core_types.Tensor]]] = None,
1711 inputs: List[List[Optional[core_types.Tensor]]] = None,
1868 inputs: Optional[List[core_types.Tensor]] = None,
1954 inputs: List[List[Optional[core_types.Tensor]]] = None,
2016 inputs: List[List[Optional[core_types.Tensor]]] = None,
2173 inputs: Optional[List[core_types.Tensor]] = None,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | propagator_debug_utils.h | 22 class Tensor; variable
|
| D | eval_const_tensor.h | 28 class Tensor; variable
|
| D | step_stats_collector.h | 39 class Tensor; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.h | 22 class Tensor; variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.cc | 258 Tensor::Tensor(cl_mem memory, bool memory_owner, const BHWC& shape, in Tensor() function in tflite::gpu::cl::Tensor
266 Tensor::Tensor(cl_mem memory, bool memory_owner, const BHWDC& shape, in Tensor() function in tflite::gpu::cl::Tensor
274 Tensor::Tensor(cl_mem memory, bool memory_owner, cl_mem image_buffer_memory, in Tensor() function in tflite::gpu::cl::Tensor
282 Tensor::Tensor(cl_mem memory, bool memory_owner, cl_mem image_buffer_memory, in Tensor() function in tflite::gpu::cl::Tensor
290 Tensor::Tensor(Tensor&& tensor) in Tensor() function in tflite::gpu::cl::Tensor
|
| D | tensor.h | 42 Tensor() in Tensor() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.h | 26 class Tensor; variable
|
| /external/tensorflow/tensorflow/python/types/ |
| D | core.py | 42 class Tensor(object): class
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 459 class Tensor; variable
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 4045 def func(x: ops.Tensor):
4066 def func(*args: ops.Tensor):
4094 def func(t: ops.Tensor, **kwargs: ops.Tensor):
4114 def func(t: ops.Tensor, *args: ops.Tensor, **kwargs: ops.Tensor):
4134 def func(t: ops.Tensor, i: int = 1, **kwargs): # pylint: disable=bad-whitespace
4148 def func(x, y: ops.Tensor, z: int):
4164 def func(x, y, *args: ops.Tensor):
4179 def func(x, y, *args, **kwargs: ops.Tensor):
4199 x: ops.Tensor = 0, # pylint:disable=bad-whitespace
4201 **kwargs: ops.Tensor):
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | batch_util.cc | 45 Status HandleElementToSlice(const Tensor& /* element */, T* src, T* dest, in HandleElementToSlice()
79 Status HandleElementToSlice<ResourceHandle>(const Tensor& /* element */, in HandleElementToSlice()
88 Status HandleElementToSlice<Eigen::half>(const Tensor& /* element */, in HandleElementToSlice()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 103 tensor: framework_ops.Tensor,
360 inputs: Optional[Sequence[framework_ops.Tensor]] = None,
520 framework_ops.Tensor]] = None,
541 inputs: Optional[Sequence[framework_ops.Tensor]] = None,
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_data.cc | 25 Status LocalTensorHandleData::Tensor(const tensorflow::Tensor** t) const { in Tensor() function in tensorflow::LocalTensorHandleData
|