| /external/tensorflow/tensorflow/python/data/util/ |
| D | sparse.py | 40 def as_dense_shapes(shapes, classes): argument
78 def deserialize_sparse_tensors(tensors, types, shapes, classes): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_slice_dataset_op_test.cc | 41 DataTypeVector dtypes, std::vector<PartialTensorShape> shapes, in CreateTensorSliceDatasetKernel()
141 std::vector<PartialTensorShape> shapes; in TEST_F() local
211 std::vector<PartialTensorShape> shapes = {PartialTensorShape({2}), in TEST_F() local
240 std::vector<PartialTensorShape> shapes; in TEST_F() local
283 std::vector<PartialTensorShape> shapes; in TEST_F() local
327 std::vector<PartialTensorShape> shapes; in TEST_F() local
363 std::vector<PartialTensorShape> shapes = {PartialTensorShape({2}), in TEST_F() local
398 std::vector<PartialTensorShape> shapes; in TEST_F() local
447 std::vector<PartialTensorShape> shapes; in TEST_F() local
495 std::vector<PartialTensorShape> shapes = {PartialTensorShape({2}), in TEST_F() local
[all …]
|
| D | dataset_test_base.cc | 40 const std::vector<PartialTensorShape>& shapes, in CreateTensorSliceDatasetKernel()
62 std::vector<PartialTensorShape> shapes; in CreateTensorSliceDataset() local
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 682 def _shapes(tensor_list_list, shapes, enqueue_many): argument
762 enqueue_many=False, shapes=None, dynamic_pad=False, argument
802 enqueue_many=False, shapes=None, dynamic_pad=False, argument
838 shapes=None, allow_smaller_final_batch=False, argument
882 enqueue_many=False, shapes=None, argument
931 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1030 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1087 shapes=None, dynamic_pad=False, allow_smaller_final_batch=False, argument
1197 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1253 num_threads=1, seed=None, enqueue_many=False, shapes=None, argument
[all …]
|
| D | saver_test_utils.py | 88 def restore(self, restore_tensors, shapes): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bcast_ops.cc | 42 absl::InlinedVector<BCast::Vec, 2> shapes; in Compile() local
91 absl::InlinedVector<BCast::Vec, 4> shapes; in Compile() local
|
| D | pack_op.cc | 47 std::vector<TensorShape> shapes; in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 59 def _as_shape_list(shapes, argument
138 def __init__(self, dtypes, shapes, names, queue_ref): argument
243 def shapes(self): member in QueueBase
636 shapes=None, argument
721 shapes=None, argument
784 shapes, argument
857 shapes=None, argument
919 def __init__(self, types, shapes=None, shared_name=None, name="barrier"): argument
1497 shapes=None, argument
1544 def shapes(self): member in BaseStagingArea
[all …]
|
| /external/libaom/libaom/test/ |
| D | av1_nn_predict_test.cc | 180 static const NN_CONFIG shapes[] = { variable
190 void NnPredictTest::RunNnPredictTest_all(const NN_CONFIG *const shapes, in RunNnPredictTest_all()
195 void NnPredictTest::RunNnPredictSpeedTest_all(const NN_CONFIG *const shapes, in RunNnPredictSpeedTest_all()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_infeed_ops.cc | 56 std::vector<PartialTensorShape> shapes; in __anon9e8f715f0102() local
|
| D | tpu_outfeed_ops.cc | 52 std::vector<PartialTensorShape> shapes; in __anon71ff5a600102() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bcast_ops.cc | 34 gtl::InlinedVector<BCast::Vec, 4> shapes; in Compute() local
83 gtl::InlinedVector<BCast::Vec, 4> shapes; in Compute() local
|
| D | sparse_concat_op.cc | 64 OpInputList shapes; in Compute() local
|
| /external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/ |
| D | ignite_dataset_ops.cc | 55 std::vector<PartialTensorShape>* shapes) { in SchemaToShapes()
180 std::vector<PartialTensorShape> shapes; in MakeDataset() local
|
| D | ignite_dataset.cc | 28 std::vector<PartialTensorShape> shapes) in IgniteDataset()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | test_util.cc | 30 std::vector<PartialTensorShape> shapes; in ShapeAnnotationsMatch() local
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | bucket_ops.py | 70 shapes=None, argument
310 shapes=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree_test.cc | 518 std::vector<xla::Shape> shapes(fan_out, shape); in BM_Construct() local
532 std::vector<xla::Shape> shapes(fan_out, shape); in BM_ConstructUnowned() local
546 std::vector<xla::Shape> shapes(fan_out, shape); in BM_Copy() local
562 std::vector<xla::Shape> shapes(fan_out, shape); in BM_Move() local
578 std::vector<xla::Shape> shapes(fan_out, shape); in BM_ForEach() local
595 std::vector<xla::Shape> shapes(fan_out, shape); in BM_Iterate() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 60 std::vector<TensorShapeProto>* shapes) { in ComputeOutputTensorShapes()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor.h | 81 const RuntimeShape* const* shapes() const { return all_shape_ptr_.data(); } in shapes() function
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cholesky_op_test.py | 269 shapes, argument
318 shapes = [ variable in CholeskyBenchmark
|
| D | matrix_band_part_op_test.py | 95 shapes = [ variable in MatrixBandPartBenchmark
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model.py | 193 def set_tensor_shapes(tensors, shapes): argument
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 697 auto shapes = config->mutable_program_shape(); in TEST() local
750 auto shapes = config->mutable_program_shape(); in TEST() local
820 auto shapes = config->mutable_program_shape(); in TEST() local
871 auto shapes = config->mutable_program_shape(); in TEST() local
917 auto shapes = config->mutable_program_shape(); in TEST() local
955 auto shapes = config->mutable_program_shape(); in TEST() local
1007 auto shapes = config->mutable_program_shape(); in TEST() local
1063 auto shapes = config->mutable_program_shape(); in TEST() local
1113 auto shapes = config->mutable_program_shape(); in TEST() local
1193 auto shapes = config->mutable_program_shape(); in TEST() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_sharding.py | 219 def get_unsharded_shape(self, shapes): argument
|