Searched refs:buffer_value (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_interpreter_test.cc | 278 int buffer_value = cur_tensor->is_variable ? 2 : 1; in TF_LITE_MICRO_TEST() local
283 buffer[j] = static_cast<int32_t>(buffer_value); in TF_LITE_MICRO_TEST()
290 buffer[j] = static_cast<uint8_t>(buffer_value); in TF_LITE_MICRO_TEST()
307 int buffer_value = cur_tensor->is_variable ? 0 : 1; in TF_LITE_MICRO_TEST() local
312 TF_LITE_MICRO_EXPECT_EQ(buffer_value, buffer[j]); in TF_LITE_MICRO_TEST()
319 TF_LITE_MICRO_EXPECT_EQ(buffer_value, buffer[j]); in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 528 [this](const BufferValue& buffer_value) -> int64 { in RunHloPassesAndBufferAssignement() argument
529 return GpuCompiler::GetSizeOfShape(buffer_value.shape(), pointer_size_); in RunHloPassesAndBufferAssignement()
571 [pointer_size](const BufferValue& buffer_value) -> int64 { in CompileModuleToLlvmIrImpl() argument
572 return GpuCompiler::GetSizeOfShape(buffer_value.shape(), pointer_size); in CompileModuleToLlvmIrImpl()
|
| D | BUILD | 1553 "//tensorflow/compiler/xla/service:buffer_value",
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.cc | 579 for (const HloValue* buffer_value : buffer.values()) { in AddAssignment() local
580 CHECK(!allocation_index_for_value_.contains(buffer_value)) in AddAssignment()
581 << "BufferValue " << buffer_value << " already has an allocation."; in AddAssignment()
582 allocation->AddAssignment(*buffer_value, offset, size); in AddAssignment()
583 allocation_index_for_value_[buffer_value] = allocation->index(); in AddAssignment()
|
| D | BUILD | 1166 ":buffer_value",
1345 ":buffer_value",
1417 ":buffer_value",
1440 ":buffer_value",
3131 name = "buffer_value",
3132 srcs = ["buffer_value.cc"],
3133 hdrs = ["buffer_value.h"],
3151 ":buffer_value",
3164 ":buffer_value",
3182 ":buffer_value",
[all …]
|