| /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/python/training/ |
| D | input.py | 681 def _shapes(tensor_list_list, shapes, enqueue_many): argument
761 enqueue_many=False, shapes=None, dynamic_pad=False, argument
801 enqueue_many=False, shapes=None, dynamic_pad=False, argument
837 shapes=None, allow_smaller_final_batch=False, argument
881 enqueue_many=False, shapes=None, argument
930 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1029 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1086 shapes=None, dynamic_pad=False, allow_smaller_final_batch=False, argument
1196 enqueue_many=False, shapes=None, dynamic_pad=False, argument
1252 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/lite/kernels/ |
| D | cpu_backend_gemm_test.cc | 523 void TestRandomGemms(const std::vector<std::tuple<int, int, int>>& shapes) { in TestRandomGemms() argument
553 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
561 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
569 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
577 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
585 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
596 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
605 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
615 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
623 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
|
| /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
785 shapes=None, argument
862 shapes, argument
935 shapes=None, argument
997 def __init__(self, types, shapes=None, shared_name=None, name="barrier"): argument
1613 shapes=None, argument
[all …]
|
| /external/libaom/libaom/test/ |
| D | av1_nn_predict_test.cc | 182 static const NN_CONFIG shapes[] = { variable
192 void NnPredictTest::RunNnPredictTest_all(const NN_CONFIG *const shapes, in RunNnPredictTest_all()
197 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 __anonc64ed8e10102() local
|
| D | tpu_outfeed_ops.cc | 52 std::vector<PartialTensorShape> shapes; in __anon91abb2220102() 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/kotlinx.coroutines/ui/kotlinx-coroutines-android/animation-app/app/src/main/java/org/jetbrains/kotlinx/animation/ |
| D | Animation.kt | 33 private var shapes = NO_SHAPES variable
64 private val shapes = MutableLiveData<Set<AnimatedShape>>() constant
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | test_util.cc | 30 std::vector<PartialTensorShape> shapes; in ShapeAnnotationsMatch() local
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | toco_convert.py | 34 shapes, 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/compiler/xrt/tests/ |
| D | raw_api_test.cc | 801 auto shapes = config->mutable_program_shape(); in TEST() local
871 auto shapes = config->mutable_program_shape(); in TEST() local
993 auto shapes = config->mutable_program_shape(); in TEST() local
1046 auto shapes = config->mutable_program_shape(); in TEST() local
1116 auto shapes = config->mutable_program_shape(); in TEST() local
1168 auto shapes = config->mutable_program_shape(); in TEST() local
1214 auto shapes = config->mutable_program_shape(); in TEST() local
1252 auto shapes = config->mutable_program_shape(); in TEST() local
1304 auto shapes = config->mutable_program_shape(); in TEST() local
1360 auto shapes = config->mutable_program_shape(); in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cholesky_op_test.py | 249 shapes, argument
298 shapes = [ variable in CholeskyBenchmark
|
| D | matrix_band_part_op_test.py | 95 shapes = [ variable in MatrixBandPartBenchmark
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 162 llvm::SmallVector<Shape, 4> shapes; in TypeToShape() local
|
| /external/tensorflow/tensorflow/lite/experimental/ruy/ |
| D | test_fast.cc | 67 const int shapes[][3] = { in TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_tf2xla.cc | 54 std::vector<std::vector<int>> shapes; in ConvertInputInfo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | mlir_roundtrip_flags.cc | 52 absl::string_view shapes, in ParseInputArrayInfo()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_sharding.py | 218 def get_unsharded_shape(self, shapes): argument
|