Home
last modified time | relevance | path

Searched refs:y_loc (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h115 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc,
259 llvm::Value* y_loc, llvm::Value* x_loc,
266 llvm::Value* y_loc, llvm::Value* x_loc,
273 llvm::Value* y_loc, llvm::Value* x_loc,
Dir_emitter_unnested.cc96 std::function<void(const llvm_ir::IrArray::Index& index, llvm::Value* y_loc,
2339 llvm::Value* y_loc = builder->CreateAdd(y_indvar, y); in EmitFullElementalTile() local
2348 emit_elem_function(source_idx_y_x, y_loc, x_loc, j); in EmitFullElementalTile()
2396 llvm::Value* y_loc = builder->CreateAdd(y_indvar, y); in EmitPartialElementalTile() local
2398 builder->CreateICmpULT(y_loc, tile_height), [&] { in EmitPartialElementalTile()
2402 y_loc, x_loc, j); in EmitPartialElementalTile()
2457 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForCopy() argument
2464 {b_.getInt64(0), x_loc, y_loc}), in EmitTileElementForCopy()
2487 const KernelCodegenInfo* kernel_info, llvm::Value* y_loc, in EmitTileElementForFusion() argument
2496 tiled_param_info->set_y(y_loc); in EmitTileElementForFusion()
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers_test.py3582 def _SpatialSoftmax(self, x_loc, y_loc, height, width, batch_size, nchannels): argument
3587 y_lin = np.expand_dims(np.array([width_lin[i] for i in y_loc]), 1)
3630 y_loc = [avg_y]
3632 np_keypoints = self._SpatialSoftmax(x_loc, y_loc, height, width, batch_size,
3652 x_loc, y_loc = zip(*edges)
3654 np_features[:, x_loc[c], y_loc[c], c] = 100.
3656 np_keypoints = self._SpatialSoftmax(x_loc, y_loc, height, width, batch_size,
3681 y_loc = [10, 9]
3683 np_features1[:, x_loc[c], y_loc[c], c] = 100.
3684 np_features2[:, x_loc[c], y_loc[c], c] = 100.
[all …]