Searched refs:shape_str (Results 1 – 8 of 8) sorted by relevance
| /external/libaom/libaom/test/ |
| D | av1_nn_predict_test.cc | 83 char shape_str[32] = { 0 }; in RunNnPredictTest() local
84 snprintf(shape_str, sizeof(shape_str), "%d", shape->num_inputs); in RunNnPredictTest()
86 snprintf(&shape_str[strlen(shape_str)], in RunNnPredictTest()
87 sizeof(shape_str) - strlen(shape_str), "x%d", in RunNnPredictTest()
89 snprintf(&shape_str[strlen(shape_str)], sizeof(shape_str) - strlen(shape_str), in RunNnPredictTest()
128 << shape_str << ")"; in RunNnPredictTest()
134 << shape_str << ")"; in RunNnPredictTest()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shaped_buffer.cc | 97 string shape_str; in ToString() local
99 shape_str = "tuple"; in ToString()
101 shape_str = ShapeUtil::HumanStringWithLayout(subshape); in ToString()
106 memory.size(), shape_str); in ToString()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader.cc | 305 string shape_str; in DebugString() local
308 strings::StrAppend(&shape_str, e.first, " (", in DebugString()
314 strings::StrAppend(&shape_str, ", ", num_slices, " slices"); in DebugString()
316 strings::StrAppend(&shape_str, "\n"); in DebugString()
319 return shape_str; in DebugString()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | toco_convert.py | 51 shape_str = ":".join([",".join(str(y) for y in x) for x in shapes if x])
62 if shape_str:
63 s += (" --input_shapes=%s" % shape_str)
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | parameterized_truncated_normal_op_test.py | 490 shape_str = "-".join(map(str, shape))
492 name="parameterized_shape" + shape_str,
496 name="naive_shape" + shape_str, iters=num_iters, wall_time=n_dt)
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | collective_ops_test.cc | 68 std::string shape_str = shape.ToString(/*print_layout=*/false); in MakeCrsModule() local
69 if (shape_str == "f32[1]") { in MakeCrsModule()
79 {{"SHAPE", shape_str}, in MakeCrsModule()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 1127 string shape_str; in DebugString() local
1134 strings::StrAppend(&shape_str, key(), " (", DataType_Name(entry.dtype()), in DebugString()
1136 strings::StrAppend(&shape_str, "\n"); in DebugString()
1138 return shape_str; in DebugString()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 672 auto shape_str = ShapeString(shape_ptr, shape_length); in __xla_cpu_runtime_AllReduce() local
673 VLOG(2) << "All-reduce input/output shape : " << shape_str; in __xla_cpu_runtime_AllReduce()
|