Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/compiler/xla/client/ | |
compile_only_client.h | 44 std::vector<const Shape*> argument_layouts; member in struct:xla::CompileOnlyClient::AotXlaComputationInstance |
/external/tensorflow/tensorflow/compiler/xla/service/ | |
compile_only_service.h | 44 std::vector<const Shape*> argument_layouts; member in struct:xla::CompileOnlyService::AotXlaComputationInstance |
/external/tensorflow/tensorflow/compiler/xla/tools/ | |
replay_computation.cc | 96 std::vector<Shape> argument_layouts; local 97 argument_layouts.reserve( 102 argument_layouts.push_back(Shape(param)); 103 argument_layout_ptrs.push_back(&argument_layouts.back()); |