| /external/tensorflow/tensorflow/core/kernels/ |
| reshape_util.h | 19 #include "tensorflow/core/framework/tensor.h"
25 void Reshape(OpKernelContext *context, const Tensor &input_indices_in,
26 const Tensor &input_shape_in, const Tensor &target_shape_in,
|
| batch_util.h | 18 #include "tensorflow/core/framework/tensor.h"
30 Status CopyElementToSlice(Tensor element, Tensor* parent, int64 index);
33 Status CopySliceToElement(const Tensor& parent, Tensor* element, int64 index);
35 // Zero-initializes the tensor `element` using the scalar stored in `padding`.
37 Status SetElementZero(Tensor* element, const Tensor& padding);
42 Status CopyElementToLargerSlice(const Tensor& element, Tensor* parent
[all...] |
| cast_op_impl.h | 67 return [](OpKernelContext* ctx, const Tensor& inp, Tensor* out) { \
74 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
77 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
80 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
83 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
[all...] |
| initializable_lookup_table.h | 30 // Performs batch lookups, for every element in the key tensor, Find returns
31 // the corresponding value into the values tensor.
44 Status Find(OpKernelContext* ctx, const Tensor& keys, Tensor* values,
45 const Tensor& default_value) final;
48 Status Insert(OpKernelContext* ctx, const Tensor& keys,
49 const Tensor& values) final {
60 Status ImportValues(OpKernelContext* ctx, const Tensor& keys,
61 const Tensor& values) final {
107 // Returns a tensor that contains the current batch of 'key' values
[all...] |
| cast_op_impl_bool.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| cast_op_impl_double.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| cast_op_impl_int16.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| cast_op_impl_int32.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| cast_op_impl_int64.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| cast_op_impl_int8.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| cast_op_impl_uint16.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| cast_op_impl_uint8.cc | 23 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
30 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
39 std::function<void(OpKernelContext*, const Tensor&, Tensor*)>
|
| inplace_ops_functor.h | 19 #include "tensorflow/core/framework/tensor.h"
26 Status DoParallelConcat(const Device& device, const Tensor& value, int32 loc,
27 Tensor* output);
|
| nth_element_op.h | 29 void operator()(OpKernelContext* context, const Tensor& input_tensor,
30 Tensor& output_tensor, int n);
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_util.h | 19 #include "tensorflow/core/framework/tensor.h"
23 namespace tensor { namespace in namespace:tensorflow
25 // DeepCopy returns a tensor whose contents are a deep copy of the
30 // REQUIRES: 'other' must be a Tensor of a copy-able type if
32 Tensor DeepCopy(const Tensor& other);
34 // Concatenates 'tensors' into a single tensor, along their 0th dimension.
39 // REQUIRES: Each member of 'tensors' must be a Tensor of a copy-able type if it
41 Status Concat(const gtl::ArraySlice<Tensor>& tensors,
42 Tensor* result) TF_MUST_USE_RESULT
[all...] |
| lookup_interface.h | 20 #include "tensorflow/core/framework/tensor.h"
36 // Performs batch lookups, for every element in the key tensor, Find returns
37 // the corresponding value into the values tensor.
50 virtual Status Find(OpKernelContext* ctx, const Tensor& keys, Tensor* values,
51 const Tensor& default_value) = 0;
53 // Inserts elements into the table. Each element of the key tensor is
54 // associated with the corresponding element in the value tensor.
64 virtual Status Insert(OpKernelContext* ctx, const Tensor& keys,
65 const Tensor& values) = 0
[all...] |
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| Tensors.java | 20 /** Type-safe factory methods for creating {@link org.tensorflow.Tensor} objects. */
25 * Creates a scalar String tensor using the default, UTF-8 encoding.
27 * @param data The string to put into the new scalar tensor.
29 public static Tensor<String> create(String data) {
30 return Tensor.create(data.getBytes(UTF_8), String.class);
34 * Creates a scalar String tensor using a specified encoding.
37 * @param data The string to put into the new scalar tensor.
39 public static Tensor<String> create(String data, java.nio.charset.Charset charset) {
40 return Tensor.create(data.getBytes(charset), String.class);
44 * Creates a scalar tensor containing a single {@code float} element
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| dma_helper.h | 19 #include "tensorflow/core/framework/tensor.h"
26 static bool CanUseDMA(const Tensor* t) { return t->CanUseDMA(); }
27 static const void* base(const Tensor* t) { return t->base<const void>(); }
28 static void* base(Tensor* t) { return t->base<void>(); }
29 static TensorBuffer* buffer(Tensor* t) { return t->buf_; }
30 static const TensorBuffer* buffer(const Tensor* t) { return t->buf_; }
|
| /external/tensorflow/tensorflow/contrib/tensorboard/db/ |
| summary_converter.h | 19 #include "tensorflow/core/framework/tensor.h"
25 Status AddTensorAsScalarToSummary(const Tensor& t, const string& tag,
27 Status AddTensorAsHistogramToSummary(const Tensor& t, const string& tag,
29 Status AddTensorAsImageToSummary(const Tensor& tensor, const string& tag,
30 int max_images, const Tensor& bad_color,
32 Status AddTensorAsAudioToSummary(const Tensor& tensor, const string& tag,
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| ring.cc | 36 template Status RingAllreduce<CPUDevice, int>(OpKernelContext*, const Tensor*,
37 Tensor*, Tensor*);
39 const Tensor*, Tensor*,
40 Tensor*);
41 template Status RingAllreduce<CPUDevice, float>(OpKernelContext*, const Tensor*,
42 Tensor*, Tensor*);
45 template Status RingAllgather<CPUDevice, int>(OpKernelContext*, const Tensor*,
[all...] |
| /external/eigen/unsupported/test/ |
| cxx11_tensor_math.cpp | 12 #include <Eigen/CXX11/Tensor>
14 using Eigen::Tensor;
19 Tensor<float, 1> vec1(6);
22 Tensor<float, 1> vec2 = vec1.tanh();
31 Tensor<float, 1> vec1(6);
34 Tensor<float, 1> vec2 = vec1.sigmoid();
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_util.h | 21 #include "tensorflow/core/framework/tensor.h"
34 // "tensor" is GPU-local. "dev" is the hosting GPU.
36 // which provides the Tensor.
39 // tensor is dead with an uninit value.
40 static void SetProtoFromGPU(const Tensor& tensor, Device* dev,
51 const Tensor* gpu_tensor, Tensor* cpu_tensor,
64 // For debugging purpose, given a "device" and a "tensor" allocated
65 // on the device, return a string printing each byte in the tensor
[all...] |
| /external/tensorflow/tensorflow/python/lib/core/ |
| py_func.h | 22 #include "tensorflow/core/framework/tensor.h"
45 // The C++ runtime converts outputs back to Tensor objects.
52 // Creates a numpy array in 'ret' and copies the content of tensor 't'
54 Status ConvertTensorToNdarray(const Tensor& t, PyObject** ret);
57 // Tensor in '*ret'.
58 Status ConvertNdarrayToTensor(PyObject* obj, Tensor* ret);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| shape_util.h | 21 #include "tensorflow/core/framework/tensor.h"
25 // Converts a TensorShape to a constant Tensor.
30 Tensor* shape_constant);
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| utils.h | 20 #include "tensorflow/core/framework/tensor.h"
27 float LeftProbability(const Tensor& point, const Tensor& weight, float bias,
30 float LeftProbabilityK(const Tensor& point, std::vector<int32> feature_set,
31 const Tensor& weight, float bias, int num_features,
|
