Searched refs:ShouldAllocateEmptyTensors (Results 1 – 5 of 5) sorted by relevance
38 virtual bool ShouldAllocateEmptyTensors() override final { return true; } in ShouldAllocateEmptyTensors() function
156 virtual bool ShouldAllocateEmptyTensors() { return false; } in ShouldAllocateEmptyTensors() function315 bool ShouldAllocateEmptyTensors() override { in ShouldAllocateEmptyTensors() function
737 if (shape_.num_elements() > 0 || a->ShouldAllocateEmptyTensors()) { in Tensor()751 if (shape_.num_elements() > 0 || a->ShouldAllocateEmptyTensors()) { in Tensor()
107 bool ShouldAllocateEmptyTensors() override { return false; } in ShouldAllocateEmptyTensors() function
61 bool ShouldAllocateEmptyTensors() override { return true; } in ShouldAllocateEmptyTensors() function in tensorflow::XlaCompilationAllocator