HomeSort by relevance Sort by last modified time
    Searched refs:x_tensor (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
direct_session_with_tracking_alloc_test.cc 56 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
57 test::FillValues<float>(&x_tensor, {1, 1});
58 Node* x = test::graph::Constant(&graph, x_tensor);
174 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
175 test::FillValues<float>(&x_tensor, {1, 1});
176 Node* x = test::graph::Constant(&graph, x_tensor);
265 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
266 test::FillValues<float>(&x_tensor, {1, 1});
267 Node* x = test::graph::Constant(&graph, x_tensor);
327 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}))
    [all...]
direct_session_test.cc 93 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
94 test::FillValues<float>(&x_tensor, {1, 1});
95 Node* x = test::graph::Constant(&graph, x_tensor);
535 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
536 x_tensor.flat<float>().setRandom();
537 Node* x = test::graph::Constant(&graph, x_tensor);
    [all...]
  /external/tensorflow/tensorflow/python/debug/lib/
debug_gradients.py 175 with respect to the any paths that include the `x_tensor`s, the gradient
308 def gradient_tensor(self, x_tensor):
312 x_tensor: (`tf.Tensor`, `tf.Variable` or `str`) The x-tensor object or its
320 TypeError: If `x_tensor` is not a `tf.Tensor`, `tf.Variable` or `str`.
321 LookupError: If the `x_tensor` has not been registered with a gradient
324 x_tensor_name = self._get_tensor_name(x_tensor)
347 "x_tensor must be a str or tf.Tensor or tf.Variable, "
372 def gradient_values_from_dump(grad_debugger, x_tensor, dump):
377 x_tensor: (`tf.Tensor`, `tf.Variable` or `str`) The x-tensor object or its
383 If this `GradientsDebugger` instance has the gradient tensor of `x_tensor`
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
gru_ops.cc 34 const Tensor* x_tensor = nullptr; variable
35 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor));
52 const int64 batch_size = x_tensor->dim_size(0);
53 const int64 input_size = x_tensor->dim_size(1);
152 ctx, device, x_tensor->matrix<T>(), h_prev_tensor->matrix<T>(),
177 const Tensor* x_tensor = nullptr; variable
178 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor));
207 const int64 batch_size = x_tensor->dim_size(0);
208 const int64 input_size = x_tensor->dim_size(1);
360 ctx, device, x_tensor->matrix<T>(), h_prev_tensor->matrix<T>()
    [all...]
lstm_ops.cc 244 const Tensor* x_tensor = nullptr; variable
245 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor));
268 const int64 batch_size = x_tensor->dim_size(0);
269 const int64 input_size = x_tensor->dim_size(1);
359 x_tensor->matrix<T>(), cs_prev_tensor->matrix<T>(),
425 const Tensor* x_tensor = nullptr; variable
426 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor));
473 const int64 batch_size = x_tensor->dim_size(0);
474 const int64 input_size = x_tensor->dim_size(1);
643 ctx, device, use_peephole_, x_tensor->matrix<T>()
962 const Tensor x_tensor = slicer.InputSlice(*x, t, "x"); variable
1220 const Tensor& x_tensor = slicer.InputSlice(*x, t, "x"); variable
    [all...]
  /external/tensorflow/tensorflow/compiler/tests/
listdiff_op_test.py 37 x_tensor = ops.convert_to_tensor(x, dtype=dtype)
41 x_tensor, y_tensor, out_idx=index_dtype)
  /external/tensorflow/tensorflow/core/platform/
device_tracer_test.cc 63 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
64 test::FillValues<float>(&x_tensor, {1, 1});
65 Node* x = test::graph::Constant(&graph, x_tensor);
  /external/tensorflow/tensorflow/python/kernel_tests/
listdiff_op_test.py 46 x_tensor = ops.convert_to_tensor(x, dtype=dtype)
48 out_tensor, idx_tensor = diff_func(x_tensor, y_tensor,
  /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
mixture_test.py 723 x_tensor = array_ops.placeholder(shape=(), dtype=dtypes.float32)
732 x_cdf_tf = mixture_tf.cdf(x_tensor)
733 x_log_cdf_tf = mixture_tf.log_cdf(x_tensor)
738 [x_cdf_tf, x_log_cdf_tf], feed_dict={x_tensor: x_feed})
772 x_tensor = array_ops.placeholder(shape=psize, dtype=dtypes.float32)
778 x_cdf_tf = mixture_tf.cdf(x_tensor)
779 x_log_cdf_tf = mixture_tf.log_cdf(x_tensor)
785 feed_dict={x_tensor: x_feed})
    [all...]
  /external/tensorflow/tensorflow/core/distributed_runtime/
master_test.cc 373 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
374 test::FillValues<float>(&x_tensor, {0, 0});
375 Node* x_node = test::graph::Constant(&graph, x_tensor);
  /external/tensorflow/tensorflow/python/ops/
image_ops_test.py     [all...]
  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_session_test.cc 302 Tensor x_tensor(DT_FLOAT, TensorShape({2, 1}));
303 test::FillValues<float>(&x_tensor, {0, 0});
304 Node* x = test::graph::Constant(&graph, x_tensor);
    [all...]
  /external/tensorflow/tensorflow/python/data/kernel_tests/
map_test.py 691 def _map_fn(x_tensor):
695 _map_py_func, [x_tensor], [dtypes.int64, dtypes.float64])
    [all...]

Completed in 1155 milliseconds