Searched refs:kTensorShapes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_memory_profile_test.cc | 55 {StatType::kTensorShapes, "[3, 3, 512, 512]"}}); in TEST()
70 {StatType::kTensorShapes, ""}}); in TEST()
86 {StatType::kTensorShapes, "[1, 2]"}}); in TEST()
|
| D | xplane_to_step_events.cc | 215 case StatType::kTensorShapes: in ConvertDeviceTraceXLineToStepEvents()
|
| D | xplane_to_memory_profile.cc | 155 case StatType::kTensorShapes: in GenerateMemoryProfile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_utils_test.cc | 348 const TensorShape kTensorShapes[] = { in TEST_P() local
352 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P()
453 const TensorShape kTensorShapes[] = {{3}, {4, 5}, {6, 7, 8}, {9, 10, 11, 12}}; in TEST_P() local
455 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P()
491 const TensorShape kTensorShapes[] = {{3}, {4, 5}, {6, 7, 8}, {9, 10, 11, 12}}; in TEST_P() local
495 for (const TensorShape& test_shape : kTensorShapes) { in TEST_P()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | gpu_event_stats.cc | 40 case StatType::kTensorShapes: in GpuEventStats()
|
| D | cost_utils.cc | 97 case StatType::kTensorShapes: in Predict()
|
| D | xplane_schema.cc | 208 {"shape", kTensorShapes}, in GetStatTypeMap()
|
| D | xplane_schema.h | 198 kTensorShapes, enumerator
|