Home
last modified time | relevance | path

Searched refs:from_tensor (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_from_tensor_op_test.py40 RaggedTensor.from_tensor(dt), [[5, 7, 0], [0, 3, 0], [6, 0, 0]])
43 RaggedTensor.from_tensor(dt, lengths=[1, 0, 3]), [[5], [], [6, 0, 0]])
46 RaggedTensor.from_tensor(dt, padding=0), [[5, 7], [0, 3], [6]])
52 RaggedTensor.from_tensor(dt_3d, lengths=([2, 0, 3], [1, 1, 2, 0, 1])),
315 rt = RaggedTensor.from_tensor(dt, lengths, padding, ragged_rank)
317 rt = RaggedTensor.from_tensor(dt, lengths, padding)
331 rt = RaggedTensor.from_tensor(dt, ragged_rank=ragged_rank)
443 rt = RaggedTensor.from_tensor(dt, lengths, padding)
504 self.assertRaisesRegexp(error[0], error[1], RaggedTensor.from_tensor, dt,
Dragged_conversion_ops.py24 def from_tensor(tensor, lengths=None, padding=None, ragged_rank=1, name=None): function
28 return ragged_tensor.RaggedTensor.from_tensor(tensor, lengths, padding,
Dragged_string_ops.py100 ragged_conversion_ops.from_tensor(input_tensor), output_encoding,
358 input = ragged_conversion_ops.from_tensor(
362 ragged_conversion_ops.from_tensor(
Dragged_tensor_shape.py161 def from_tensor(cls, rt_input): member in RaggedTensorDynamicShape
491 src_shape = RaggedTensorDynamicShape.from_tensor(rt_input)
512 ragged_conversion_ops.from_tensor(
515 rt_input = ragged_conversion_ops.from_tensor(
554 rt_input = ragged_conversion_ops.from_tensor(rt_input, ragged_rank=1)
Dragged_array_ops.py135 data = ragged_conversion_ops.from_tensor(
195 mask = ragged_conversion_ops.from_tensor(
526 rt_input = ragged_conversion_ops.from_tensor(rt_input)
Dragged_concat_ops.py171 rt_inputs[i] = ragged_conversion_ops.from_tensor(
288 rt_input = ragged_conversion_ops.from_tensor(rt_input)
Dragged_batch_gather_ops.py100 params = ragged_conversion_ops.from_tensor(params, ragged_rank=1)
Dragged_batch_gather_with_default_op.py150 params_shape = ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(
Dragged_gather_ops.py199 indices = ragged_conversion_ops.from_tensor(
Dragged_dispatch.py204 ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(x),
205 ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(y))
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc74 const AllocatorAttributes& from_alloc_attr, const Tensor* from_tensor, in PostToPeer() argument
78 buf_rendezvous_.ProvideBuf(key, from_device, from_device_ctx, from_tensor, in PostToPeer()
Dbase_collective_executor.h136 const Tensor* from_tensor, in PostToPeer() argument
140 from_device_ctx, from_alloc_attr, from_tensor, in PostToPeer()
Dcollective_rma_local.h51 const Tensor* from_tensor,
Dgraph_execution_state.cc254 NodeBuilder::NodeOut from_tensor) in TensorConnectionPruneRewrite() argument
256 from_tensor_(std::move(from_tensor)) {} in TensorConnectionPruneRewrite()
507 TensorId from_id(ParseTensorName(tensor_connection.from_tensor())); in PruneGraph()
628 item.fetch.push_back(tensor_connection.from_tensor()); in OptimizeGraph()
Dtest_collective_executor_mgr.h45 const Tensor* from_tensor, in PostToPeer() argument
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor_test.py103 from_tensor = sparse_tensor.convert_to_tensor_or_sparse_tensor(st).eval()
104 for convertee in [from_value, from_tensor]:
Dtensor_spec_test.py118 spec = tensor_spec.TensorSpec.from_tensor(zero)
129 spec_1 = tensor_spec.TensorSpec.from_tensor(unknown)
133 spec_2 = tensor_spec.TensorSpec.from_tensor(partial)
Dtensor_spec.py65 def from_tensor(cls, tensor, name=None): member in TensorSpec
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-tensor-spec.pbtxt26 name: "from_tensor"
Dtensorflow.-ragged-tensor.pbtxt79 name: "from_tensor"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-tensor-spec.pbtxt26 name: "from_tensor"
Dtensorflow.-ragged-tensor.pbtxt79 name: "from_tensor"
/external/tensorflow/tensorflow/python/keras/utils/
Dcomposite_tensor_support_test.py67 return ragged_tensor.RaggedTensor.from_tensor(
/external/tensorflow/tensorflow/python/saved_model/
Dsignature_serialization.py117 tensor_spec_signature[keyword] = tensor_spec.TensorSpec.from_tensor(
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/
Dexport_mnist_cnn.py122 hparams_spec = {name: tf.TensorSpec.from_tensor(tf.constant(value))

12