Home
last modified time | relevance | path

Searched defs:shapes (Results 1 – 25 of 71) sorted by relevance

123

/external/tensorflow/tensorflow/python/data/util/
Dsparse.py40 def as_dense_shapes(shapes, classes): argument
78 def deserialize_sparse_tensors(tensors, types, shapes, classes): argument
/external/tensorflow/tensorflow/python/training/
Dinput.py656 def _shapes(tensor_list_list, shapes, enqueue_many): argument
736 enqueue_many=False, shapes=None, dynamic_pad=False, argument
775 enqueue_many=False, shapes=None, dynamic_pad=False, argument
810 shapes=None, allow_smaller_final_batch=False, argument
854 enqueue_many=False, shapes=None, argument
899 enqueue_many=False, shapes=None, dynamic_pad=False, argument
994 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1047 shapes=None, dynamic_pad=False, allow_smaller_final_batch=False, argument
1153 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1206 num_threads=1, seed=None, enqueue_many=False, shapes=None, argument
[all …]
Dsaver_test_utils.py88 def restore(self, restore_tensors, shapes): argument
/external/tensorflow/tensorflow/docs_src/programmers_guide/
Dfaq.md191 #### How can I determine the shape of a tensor in Python?
202 #### What is the difference between `x.set_shape()` and `x = tf.reshape(x)`?
212 #### How do I build a graph that works with variable batch sizes?
/external/tensorflow/tensorflow/contrib/tpu/ops/
Dinfeed_ops.cc81 std::vector<PartialTensorShape> shapes; in __anonf661f04b0102() local
Doutfeed_ops.cc72 std::vector<PartialTensorShape> shapes; in __anonc421b6cc0102() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbcast_ops.cc41 gtl::InlinedVector<BCast::Vec, 2> shapes; in Compile() local
90 gtl::InlinedVector<BCast::Vec, 4> shapes; in Compile() local
Dpack_op.cc47 std::vector<TensorShape> shapes; in Compile() local
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_ops.py57 def _as_shape_list(shapes, argument
139 def __init__(self, dtypes, shapes, names, queue_ref): argument
244 def shapes(self): member in QueueBase
626 shapes=None, argument
715 shapes=None, argument
780 shapes, argument
856 shapes=None, argument
918 def __init__(self, types, shapes=None, shared_name=None, name="barrier"): argument
1481 shapes=None, argument
1528 def shapes(self): member in BaseStagingArea
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dbcast_ops.cc34 gtl::InlinedVector<BCast::Vec, 4> shapes; in Compute() local
83 gtl::InlinedVector<BCast::Vec, 4> shapes; in Compute() local
Dsparse_concat_op.cc64 OpInputList shapes; in Compute() local
/external/tensorflow/tensorflow/contrib/tpu/python/ops/
Dtpu_ops.py77 def infeed_dequeue_tuple(dtypes, shapes, name=None): argument
/external/tensorflow/tensorflow/contrib/training/python/training/
Dbucket_ops.py70 shapes=None, argument
310 shapes=None, argument
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_queue_dataset_op.cc51 const std::vector<PartialTensorShape>& shapes) { in PrependQueueShapeWithBatch()
69 const std::vector<PartialTensorShape>& shapes, in PrependFromQueueAndPaddedBatchDataset()
171 const auto& shapes = dataset()->shapes_; in GetNextInternal() local
253 const std::vector<PartialTensorShape>& shapes) in TensorQueue()
Dreader_dataset_ops.cc104 static std::vector<PartialTensorShape>* shapes = in output_shapes() local
338 static std::vector<PartialTensorShape>* shapes = in output_shapes() local
558 static std::vector<PartialTensorShape>* shapes = in output_shapes() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dmatrix_band_part_op_test.py91 shapes = [ variable in MatrixBandPartBenchmark
Dcholesky_op_test.py254 shapes, argument
303 shapes = [ variable in CholeskyBenchmark
Dmatrix_logarithm_op_test.py126 shapes = [ variable in MatrixLogarithmBenchmark
Ddeterminant_op_test.py163 shapes = [ variable in MatrixDeterminantBenchmark
Dmatrix_inverse_op_test.py155 shapes = [ variable in MatrixInverseBenchmark
Dmatrix_exponential_op_test.py154 shapes = [ variable in MatrixExponentialBenchmark
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dtpu_sharding.py219 def get_unsharded_shape(self, shapes): argument
/external/gemmlowp/test/
Dbenchmark_meta_gemm.cc104 double run_gemms(std::vector<Shape>* shapes) { in run_gemms()
159 void time_all(std::vector<Shape>* shapes, std::int32_t repetitions, in time_all()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc168 std::vector<TensorShape> shapes; in LoadGraphFromProto() local
322 std::vector<TensorShape> shapes; in TransformGraphToAddAggregatedInputNode() local
377 std::vector<TensorShape> shapes; in TransformGraphToAddAggregatedInputNode() local
619 std::vector<TensorShape> shapes; in IsNodeFlattenReshape() local
679 std::vector<TensorShape> shapes; in RegisterNodeWithRank() local
908 std::vector<TensorShape> shapes; in AppendNodeOutputParams() local
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DOperationBuilderTest.java191 private static void testSetAttrShapeList(Shape[] shapes) { in testSetAttrShapeList()

123