Searched defs:y_loc (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2339 llvm::Value* y_loc = builder->CreateAdd(y_indvar, y); in EmitFullElementalTile() local
2396 llvm::Value* y_loc = builder->CreateAdd(y_indvar, y); in EmitPartialElementalTile() local
2457 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForCopy()
2487 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForFusion()
2854 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForReduction()
3155 llvm::Value* x_loc, int64 /*x_iter_num*/) { in EmitKernel()
3181 int64 x_iter_num) { in EmitKernel()
3258 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitHlo021Tile()
3259 llvm::Value* x_loc, int64 x_iter_num) { in EmitHlo021Tile()
3780 llvm::Value* x_loc, int64 x_iter_num) { in EmitReductionToVector()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers_test.py | 3582 def _SpatialSoftmax(self, x_loc, y_loc, height, width, batch_size, nchannels): argument
|